diff options
author | David Gross <dgross@google.com> | 2017-10-05 14:23:39 -0700 |
---|---|---|
committer | David Gross <dgross@google.com> | 2017-10-05 18:28:00 -0700 |
commit | def0a14aa77689f12120cfb4f136eea659038cc0 (patch) | |
tree | c33d4f077b8be852d8057877ac1c43d782f44e02 /nn/runtime/ExecutionPlan.h | |
parent | d5cd96a9f7cef240851caa8d846ea88f6333cfa0 (diff) | |
download | ml-def0a14aa77689f12120cfb4f136eea659038cc0.tar.gz |
Partitioning interface changes to enable unit testing.
Bug: 63905942
Test: mma (userdebug)
ml/nn/runtime/tests (userdebug)
(no failures, logcat looks plausible)
Change-Id: Ib9a5eb75b9e4e2c3534f392d5fb0e5ba3555197d
Diffstat (limited to 'nn/runtime/ExecutionPlan.h')
-rw-r--r-- | nn/runtime/ExecutionPlan.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/nn/runtime/ExecutionPlan.h b/nn/runtime/ExecutionPlan.h index 0cc605113..1dfc33fd2 100644 --- a/nn/runtime/ExecutionPlan.h +++ b/nn/runtime/ExecutionPlan.h @@ -190,6 +190,13 @@ public: void dump() const; + // These functions are solely intended for use by unit tests of + // the partitioning algorithm. + enum class Kind { ERROR, EMPTY, SIMPLE, COMPOUND }; + Kind forTest_getKind() const; + std::shared_ptr<const Device> forTest_simpleGetDevice() const; + const std::vector<std::shared_ptr<ExecutionStep>>& forTest_compoundGetSteps() const; + private: void findSubModelOutputs(); |