| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | GatherTestImpl.cpp | 28 const armnn::TensorInfo& indicesInfo, in GatherTestImpl()
78 armnn::TensorInfo indicesInfo({ 4 }, armnn::DataType::Signed32); in Gather1dParamsTestImpl() local
112 armnn::TensorInfo indicesInfo({ 2 }, armnn::DataType::Signed32); in Gather1dParamsAxisTestImpl() local
151 armnn::TensorInfo indicesInfo({ 3 }, armnn::DataType::Signed32); in GatherMultiDimParamsTestImpl() local
184 armnn::TensorInfo indicesInfo({ 2, 3 }, armnn::DataType::Signed32); in GatherMultiDimParamsMultiDimIndicesTestImpl() local
246 armnn::TensorInfo indicesInfo({ 2, 3 }, armnn::DataType::Signed32); in GatherMultiDimParamsMultiDimIndicesAxis1TestImpl() local
319 armnn::TensorInfo indicesInfo({ 2, 3 }, armnn::DataType::Signed32); in GatherMultiDimParamsMultiDimIndicesAxis2TestImpl() local
390 armnn::TensorInfo indicesInfo({ 4 }, armnn::DataType::Signed32); in Gather1dParamsTestImpl() local
417 armnn::TensorInfo indicesInfo({ 3 }, armnn::DataType::Signed32); in GatherMultiDimParamsTestImpl() local
445 armnn::TensorInfo indicesInfo({ 2, 3 }, armnn::DataType::Signed32); in GatherMultiDimParamsMultiDimIndicesTestImpl() local
|
| D | GatherNdTestImpl.cpp | 25 const armnn::TensorInfo &indicesInfo, in GatherNdTestImpl()
74 armnn::TensorInfo indicesInfo({ 3, 1 }, armnn::DataType::Signed32); in SimpleGatherNd2dTest() local
117 armnn::TensorInfo indicesInfo({ 2, 2 }, armnn::DataType::Signed32); in SimpleGatherNd3dTest() local
178 armnn::TensorInfo indicesInfo({ 2, 2, 3, 2 }, armnn::DataType::Signed32); in SimpleGatherNd4dTest() local
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | GatherNdEndToEndTestImpl.hpp | 18 const armnn::TensorInfo& indicesInfo, in CreateGatherNdNetwork()
39 armnn::TensorInfo indicesInfo({ 2, 2 }, armnn::DataType::Signed32); in GatherNdEndToEnd() local
97 armnn::TensorInfo indicesInfo({ 2, 2, 3, 2 }, armnn::DataType::Signed32); in GatherNdMultiDimEndToEnd() local
|
| D | GatherEndToEndTestImpl.hpp | 18 const armnn::TensorInfo& indicesInfo, in CreateGatherNetwork()
40 armnn::TensorInfo indicesInfo({ 3 }, armnn::DataType::Signed32); in GatherEndToEnd() local
78 armnn::TensorInfo indicesInfo({ 2, 3 }, armnn::DataType::Signed32); in GatherMultiDimEndToEnd() local
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonGatherNdWorkload.cpp | 15 const TensorInfo& indicesInfo, in NeonGatherNdWorkloadValidate()
104 TensorInfo indicesInfo = info.m_InputTensorInfos[1]; in NeonGatherNdWorkload() local
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClGatherNdWorkload.cpp | 17 const TensorInfo& indicesInfo, in ClGatherNdWorkloadValidate()
107 TensorInfo indicesInfo = info.m_InputTensorInfos[1]; in ClGatherNdWorkload() local
|
| /external/armnn/src/backends/reference/workloads/ |
| D | Gather.cpp | 15 const TensorInfo& indicesInfo, in Gather()
|
| /external/armnn/src/armnn/test/ |
| D | OptimizerTests.cpp | 630 const armnn::TensorInfo& indicesInfo, in CreateGatherGraph()
653 armnn::TensorInfo indicesInfo({3}, DataType::Signed32); variable
665 armnn::TensorInfo indicesInfo({5}, DataType::Signed32); variable
677 armnn::TensorInfo indicesInfo({2, 2}, DataType::Signed32); variable
|
| /external/armnn/src/armnnSerializer/test/ |
| D | SerializerTests.cpp | 1322 const armnn::TensorInfo indicesInfo({ 3 }, armnn::DataType::Signed32, 0.0f, 0, true); variable
1389 const armnn::TensorInfo indicesInfo({ 3, 1 }, armnn::DataType::Signed32, 0.0f, 0, true); variable
|