| /external/android-nn-driver/test/ |
| D | Concat.cpp | 23 ConcatTestImpl(const std::vector<const TestTensor*> & inputs, in ConcatTestImpl()
25 const TestTensor & expectedOutputTensor, in ConcatTestImpl()
139 TestTensor aIn{armnn::TensorShape{1, 1, 1, 1}, {0}}; in SimpleConcatAxis0()
140 TestTensor bIn{armnn::TensorShape{1, 1, 1, 1}, {1}}; in SimpleConcatAxis0()
141 TestTensor cIn{armnn::TensorShape{1, 1, 1, 1}, {2}}; in SimpleConcatAxis0()
143 TestTensor expected{armnn::TensorShape{3, 1, 1, 1}, {0, 1, 2}}; in SimpleConcatAxis0()
150 TestTensor aIn{armnn::TensorShape{2, 1, 2, 1}, {0, 1, in ConcatAxis0NoInterleave()
152 TestTensor bIn{armnn::TensorShape{3, 1, 2, 1}, {4, 5, in ConcatAxis0NoInterleave()
155 TestTensor cIn{armnn::TensorShape{1, 1, 2, 1}, {10, 11}}; in ConcatAxis0NoInterleave()
157 TestTensor expected{armnn::TensorShape{6, 1, 2, 1}, {0, 1, in ConcatAxis0NoInterleave()
[all …]
|
| D | TestTensor.cpp | 11 hidl_vec<uint32_t> TestTensor::GetDimensions() const in GetDimensions()
22 unsigned int TestTensor::GetNumElements() const in GetNumElements()
27 const float * TestTensor::GetData() const in GetData()
|
| D | TestTensor.hpp | 14 class TestTensor class
17 TestTensor(const armnn::TensorShape & shape, in TestTensor() function in driverTestHelpers::TestTensor
|
| /external/android-nn-driver/test/1.1/ |
| D | Mean.cpp | 23 void MeanTestImpl(const TestTensor& input, in MeanTestImpl()
27 const TestTensor& expectedOutput, in MeanTestImpl()
104 TestTensor input{ armnn::TensorShape{ 4, 3, 2 },
111 TestTensor expectedOutput{ armnn::TensorShape{ 2 }, { 12.0f, 13.0f } };
118 … TestTensor input{ armnn::TensorShape{ 1, 1, 3, 2 }, { 1.0f, 1.0f, 2.0f, 2.0f, 3.0f, 3.0f } };
122 TestTensor expectedOutput{ armnn::TensorShape{ 1, 1, 1, 2 }, { 2.0f, 2.0f } };
129 TestTensor input{ armnn::TensorShape{ 4, 3, 2 },
136 TestTensor expectedOutput{ armnn::TensorShape{ 2 }, { 12.0f, 13.0f } };
143 … TestTensor input{ armnn::TensorShape{ 1, 1, 3, 2 }, { 1.0f, 1.0f, 2.0f, 2.0f, 3.0f, 3.0f } };
147 TestTensor expectedOutput{ armnn::TensorShape{ 1, 1, 1, 2 }, { 2.0f, 2.0f } };
[all …]
|
| D | Transpose.cpp | 26 void TransposeTestImpl(const TestTensor & inputs, int32_t perm[], in TransposeTestImpl()
27 const TestTensor & expectedOutputTensor, armnn::Compute computeDevice) in TransposeTestImpl()
100 TestTensor input{armnn::TensorShape{1, 2, 2, 2},{1, 2, 3, 4, 5, 6, 7, 8}};
101 TestTensor expected{armnn::TensorShape{2, 2, 2, 1},{1, 5, 2, 6, 3, 7, 4, 8}};
109 … TestTensor input{armnn::TensorShape{1, 2, 2, 3},{1, 2, 3, 11, 12, 13, 21, 22, 23, 31, 32, 33}};
110 … TestTensor expected{armnn::TensorShape{1, 3, 2, 2},{1, 11, 21, 31, 2, 12, 22, 32, 3, 13, 23, 33}};
117 TestTensor input{armnn::TensorShape{1, 2, 2, 2},{1, 2, 3, 4, 5, 6, 7, 8}};
118 TestTensor expected{armnn::TensorShape{1, 2, 2, 2},{1, 5, 2, 6, 3, 7, 4, 8}};
130 TestTensor input{armnn::TensorShape{1, 2, 2, 2},{1, 2, 3, 4, 5, 6, 7, 8}};
131 TestTensor expected{armnn::TensorShape{2, 2, 2, 1},{1, 5, 2, 6, 3, 7, 4, 8}};
[all …]
|
| /external/android-nn-driver/ |
| D | LICENSE.spdx | 723 FileName: ./test/TestTensor.cpp
730 FileName: ./test/TestTensor.hpp
|