| /external/armnn/src/armnnTfLiteParser/test/ |
| D | OutputShapeOfSqueeze.cpp | 17 unsigned int m_InputShape[4]; member
19 TfLiteParserFixture() : m_Parser( ), m_InputShape { 1, 2, 2, 1 } {} in TfLiteParserFixture()
29 … armnn::TensorInfo inputTensorInfo = armnn::TensorInfo(4, m_InputShape, armnn::DataType::Float32);
40 … armnn::TensorInfo inputTensorInfo = armnn::TensorInfo(4, m_InputShape, armnn::DataType::Float32);
51 … armnn::TensorInfo inputTensorInfo = armnn::TensorInfo(4, m_InputShape, armnn::DataType::Float32);
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | LogicalTestImpl.cpp | 134 armnn::TensorShape m_InputShape; member in __anoneb357ae10111::UnaryTestData
164 m_InputShape = { 1, 1, 1, 4 }; in SimpleUnaryTestData()
165 m_OutputShape = m_InputShape; in SimpleUnaryTestData()
184 m_InputShape = { 1, 1, 1, 4 }; in SimpleUnaryIntTestData()
185 m_OutputShape = m_InputShape; in SimpleUnaryIntTestData()
375 s_SimpleUnaryTestData.m_InputShape, in LogicalNotTest()
390 s_SimpleUnaryIntTestData.m_InputShape, in LogicalNotIntTest()
|
| D | ResizeTestImpl.cpp | 39 armnn::TensorShape m_InputShape; member
83 armnn::TensorInfo inputInfo(params.m_InputShape, ArmnnType); in ResizeTestImpl()
162 testParams.m_InputShape = { 1, 2, 4, 4 }; in ResizeBilinearNopTest()
163 testParams.m_OutputShape = testParams.m_InputShape; in ResizeBilinearNopTest()
196 testParams.m_InputShape = { 1, 2, 2, 2 }; in SimpleResizeBilinearTest()
236 testParams.m_InputShape = { 1, 2, 4, 4 }; in ResizeBilinearSqMinTest()
277 testParams.m_InputShape = { 1, 2, 3, 5 }; in ResizeBilinearMinTest()
316 testParams.m_InputShape = { 1, 2, 3, 2 }; in ResizeBilinearMagTest()
361 testParams.m_InputShape = { 1, 2, 4, 4 }; in ResizeNearestNeighborNopTest()
362 testParams.m_OutputShape = testParams.m_InputShape; in ResizeNearestNeighborNopTest()
[all …]
|
| D | StackTestImpl.cpp | 53 descriptor.m_Parameters.m_InputShape = inputTensorInfo.GetShape(); in StackTestHelper()
|
| /external/armnn/src/backends/reference/workloads/ |
| D | RefFullyConnectedWorkload.cpp | 30 , m_InputShape(info.m_InputTensorInfos[0].GetShape()) in RefFullyConnectedWorkload()
63 FullyConnected(m_InputShape, in Execute()
|
| D | RefConvolution2dWorkload.cpp | 18 , m_InputShape(info.m_InputTensorInfos[0].GetShape()) in RefConvolution2dWorkload()
59 Convolve(m_InputShape, *inputDecoder, m_OutputShape, *outputEncoder, m_FilterShape, in Execute()
|
| D | RefConvolution2dWorkload.hpp | 28 const TensorShape m_InputShape; member in armnn::RefConvolution2dWorkload
|
| D | RefFullyConnectedWorkload.hpp | 30 const TensorShape m_InputShape; member in armnn::RefFullyConnectedWorkload
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClStackWorkload.cpp | 42 int aclAxis = CalcAxis(descriptor.m_Axis, descriptor.m_InputShape.GetNumDimensions()); in ClStackWorkloadValidate()
68 …int aclAxis = CalcAxis(descriptor.m_Parameters.m_Axis, descriptor.m_Parameters.m_InputShape.GetNum… in ClStackWorkload()
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonStackWorkload.cpp | 45 int aclAxis = CalcAxis(descriptor.m_Axis, descriptor.m_InputShape.GetNumDimensions()); in NeonStackWorkloadValidate()
68 …int aclAxis = CalcAxis(descriptor.m_Parameters.m_Axis, descriptor.m_Parameters.m_InputShape.GetNum… in NeonStackWorkload()
|
| /external/armnn/src/armnn/layers/ |
| D | StackLayer.cpp | 39 const TensorShape& inputShape = m_Param.m_InputShape; in InferOutputShapes()
82 if (inputShape != m_Param.m_InputShape) in ValidateTensorShapesFromInputs()
|
| /external/armnn/src/armnn/test/ |
| D | InferOutputTests.hpp | 297 descriptor.m_InputShape = armnn::TensorShape in StackInferOutputShapeFromInputsMatchTest()
327 descriptor.m_InputShape = armnn::TensorShape in StackInferOutputShapeFromInputsNoMatchTest()
389 descriptor.m_InputShape = armnn::TensorShape in StackValidateTensorShapesFromInputsMatchTest()
415 descriptor.m_InputShape = armnn::TensorShape in StackValidateTensorShapesFromInputsNoMatchTest()
|
| D | ShapeInferenceTests.cpp | 695 descriptor.m_InputShape = { 3, 2, 3 };
|
| /external/armnn/include/armnn/ |
| D | Descriptors.hpp | 1223 , m_InputShape() in StackDescriptor()
1229 , m_InputShape(inputShape) in StackDescriptor()
1236 m_InputShape == rhs.m_InputShape; in operator ==()
1244 TensorShape m_InputShape; member
|
| /external/armnn/delegate/classic/src/ |
| D | Pack.hpp | 73 desc.m_InputShape = inputTensorInfos[0].GetShape(); in VisitPackOperator()
|
| /external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
| D | armnn_descriptors.i | 1072 armnn::TensorShape m_InputShape; member
|
| /external/armnn/python/pyarmnn/test/ |
| D | test_descriptors.py | 348 assert desc.m_InputShape.GetNumDimensions() == 0
|
| /external/armnn/src/armnn/ |
| D | SerializeLayerParameters.cpp | 493 ss << desc.m_InputShape; in Serialize()
|
| /external/armnn/src/armnnSerializer/ |
| D | Serializer.cpp | 1353 for (unsigned int i =0; i < stackDescriptor.m_InputShape.GetNumDimensions(); i++) in SerializeStackLayer()
1355 inputShape.push_back(stackDescriptor.m_InputShape[i]); in SerializeStackLayer()
|
| /external/armnn/src/armnnDeserializer/ |
| D | Deserializer.cpp | 3717 if (descriptor.m_InputShape != inputShape) in ParseStack()
3725 << descriptor.m_InputShape in ParseStack()
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadData.cpp | 952 const TensorShape& inputShape = m_Parameters.m_InputShape; in Validate()
|
| /external/armnn/src/armnnTfLiteParser/ |
| D | TfLiteParser.cpp | 3514 desc.m_InputShape = inputTensorInfo.GetShape(); in ParsePack()
|