diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2018-09-07 09:25:10 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-18 12:40:42 +0100 |
commit | 8b287c23f5141102555ba869a34397ec720a3e4f (patch) | |
tree | 4a60d9c806669aad06816414601ac532946d6878 /test/Lstm.cpp | |
parent | 77605826a353981d41f0ee346850d411770535f8 (diff) | |
download | android-nn-driver-8b287c23f5141102555ba869a34397ec720a3e4f.tar.gz |
IVGCVSW-1806 More Android NN Driver refactoring
* Changed #if defined to #ifdef
* Simplified the Android ML namespace resolution
* Fixed the relative path in some include directives
Change-Id: I46e46faff98559c8042c1a4b8b82007f462df57d
Diffstat (limited to 'test/Lstm.cpp')
-rw-r--r-- | test/Lstm.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/Lstm.cpp b/test/Lstm.cpp index fe6e5d0d..f5d26b2e 100644 --- a/test/Lstm.cpp +++ b/test/Lstm.cpp @@ -114,7 +114,7 @@ void LstmTestImpl(hidl_vec<uint32_t> inputDimensions, std::vector<float> outputValue) { auto driver = std::make_unique<ArmnnDriver>(DriverOptions(armnn::Compute::GpuAcc)); - neuralnetworks::V1_0::Model model = {}; + V1_0::Model model = {}; // Inputs: // 00: The input: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [batch_size, input_size], where @@ -198,7 +198,7 @@ void LstmTestImpl(hidl_vec<uint32_t> inputDimensions, // make the lstm operation model.operations.resize(1); - model.operations[0].type = neuralnetworks::V1_0::OperationType::LSTM; + model.operations[0].type = V1_0::OperationType::LSTM; model.operations[0].inputs = hidl_vec<uint32_t> {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22}; model.operations[0].outputs = hidl_vec<uint32_t> {23, 24, 25, 26}; |