diff options
| author | Xusong Wang <xusongw@google.com> | 2018-10-31 08:37:25 -0700 |
|---|---|---|
| committer | Miao Wang <miaowang@google.com> | 2019-01-24 18:08:06 -0800 |
| commit | e212fe32cbe8466f0e7ede768687f07faa95b8e2 (patch) | |
| tree | 1ef19c94c477bcc5d8aadb2c1aa7c2d80b5a7300 /nn/runtime/test/TestNeuralNetworksWrapper.h | |
| parent | 30b8bded4e60e44ff752248c51a0018c8ff74068 (diff) | |
| download | ml-e212fe32cbe8466f0e7ede768687f07faa95b8e2.tar.gz | |
Implement NDK interface for dynamic output shape.
Implement the following NDK interfaces
- ANeuralNetworksExecution_getOutputOperandDimensions
- ANeuralNetworksExecution_getOutputOperandRank
Bug: 73506513
Test: NeuralNetworksTest_static
Change-Id: I3e0238ec701a0bffbdb5682ee1787df198fe9816
Merged-In: I3e0238ec701a0bffbdb5682ee1787df198fe9816
(cherry picked from commit 1b10627d9b78def5e18e1c87584dcf59d6b93cac)
Diffstat (limited to 'nn/runtime/test/TestNeuralNetworksWrapper.h')
| -rw-r--r-- | nn/runtime/test/TestNeuralNetworksWrapper.h | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/nn/runtime/test/TestNeuralNetworksWrapper.h b/nn/runtime/test/TestNeuralNetworksWrapper.h index 7ae337012..6a0551b8b 100644 --- a/nn/runtime/test/TestNeuralNetworksWrapper.h +++ b/nn/runtime/test/TestNeuralNetworksWrapper.h @@ -422,6 +422,20 @@ class Execution { // computation to complete. static void setComputeUsesSynchronousAPI(bool val) { mComputeUsesSychronousAPI = val; } + Result getOutputOperandDimensions(uint32_t index, std::vector<uint32_t>* dimensions) { + uint32_t rank = 0; + Result result = static_cast<Result>( + ANeuralNetworksExecution_getOutputOperandRank(mExecution, index, &rank)); + dimensions->resize(rank); + if ((result != Result::NO_ERROR && result != Result::OUTPUT_INSUFFICIENT_SIZE) || + rank == 0) { + return result; + } + result = static_cast<Result>(ANeuralNetworksExecution_getOutputOperandDimensions( + mExecution, index, dimensions->data())); + return result; + } + private: ANeuralNetworksExecution* mExecution = nullptr; |