| /external/armnn/src/backends/backendsCommon/test/ |
| D | ConcatEndToEndTestImpl.hpp | 25 unsigned int concatAxis, in CreateConcatNetwork() argument
37 concatAxis); in CreateConcatNetwork()
60 unsigned int concatAxis = 0; in ConcatDim0EndToEnd() local
65 INetworkPtr net = CreateConcatNetwork<ArmnnType>(inputShapes, outputShape, concatAxis); in ConcatDim0EndToEnd()
124 unsigned int concatAxis = 1; in ConcatDim1EndToEnd() local
129 INetworkPtr net = CreateConcatNetwork<ArmnnType>(inputShapes, outputShape, concatAxis); in ConcatDim1EndToEnd()
186 unsigned int concatAxis = 2; in ConcatDim2EndToEnd() local
191 INetworkPtr net = CreateConcatNetwork<ArmnnType>(inputShapes, outputShape, concatAxis); in ConcatDim2EndToEnd()
247 unsigned int concatAxis = 3; in ConcatDim3EndToEnd() local
252 INetworkPtr net = CreateConcatNetwork<ArmnnType>(inputShapes, outputShape, concatAxis); in ConcatDim3EndToEnd()
|
| /external/armnn/src/armnnUtils/ |
| D | ParserHelper.cpp | 21 const unsigned int& concatAxis, in ProcessConcatInputTensorInfo() argument
38 for (unsigned int j = 0; j < concatAxis; ++j) in ProcessConcatInputTensorInfo()
43 concatDescriptor.SetViewOriginCoord(inputIndex, concatAxis, mergeDimOrigin); in ProcessConcatInputTensorInfo()
44 mergeDimOrigin += inputTensorInfo.GetShape()[concatAxis]; in ProcessConcatInputTensorInfo()
46 for (unsigned int j = concatAxis + 1; j < inputRank; ++j) in ProcessConcatInputTensorInfo()
|
| D | ParserHelper.hpp | 18 const unsigned int& concatAxis,
|
| /external/armnn/delegate/classic/src/ |
| D | Control.hpp | 26 const unsigned int concatAxis, in SetupConcatViewOrigin() argument
39 for (unsigned int j = 0; j < concatAxis; ++j) in SetupConcatViewOrigin()
44 concatDescriptor.SetViewOriginCoord(inputIndex, concatAxis, mergeDimOrigin); in SetupConcatViewOrigin()
45 mergeDimOrigin += inputTensorInfo.GetShape()[concatAxis]; in SetupConcatViewOrigin()
47 for (unsigned int j = concatAxis + 1; j < inputRank; ++j) in SetupConcatViewOrigin()
|
| /external/armnn/src/armnn/layers/ |
| D | ConcatLayer.cpp | 52 uint32_t concatAxis = m_Param.GetConcatAxis(); in CreateTensors() local
55 … && ((concatAxis == numberOfDimensions - 1) || (concatAxis == numberOfDimensions - 2)); in CreateTensors()
|
| /external/armnn/src/backends/tosaReference/test/ |
| D | TosaRefLayerSupportTests.cpp | 75 unsigned int concatAxis = 1; variable
76 descriptor = CreateDescriptorForConcatenation(shapes.begin(), shapes.end(), concatAxis);
101 unsigned int concatAxis = 1; variable
102 descriptor = armnn::CreateDescriptorForConcatenation(shapes.begin(), shapes.end(), concatAxis);
|
| /external/armnn/src/backends/tosaCommon/test/ |
| D | OneToOneMappingTests.cpp | 69 unsigned int concatAxis = 1; variable
70 descriptor.SetConcatAxis(concatAxis);
73 concatAxis);
99 unsigned int concatAxis = 1; variable
100 descriptor.SetConcatAxis(concatAxis);
104 concatAxis);
|
| /external/armnn/src/armnn/ |
| D | Descriptors.cpp | 158 void OriginsDescriptor::SetConcatAxis(unsigned int concatAxis) in SetConcatAxis() argument
160 m_ConcatAxis = concatAxis; in SetConcatAxis()
|
| /external/android-nn-driver/test/ |
| D | Concat.cpp | 24 int32_t concatAxis, in ConcatTestImpl() argument
41 AddIntOperand<HalPolicy>(model, concatAxis); in ConcatTestImpl()
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonCreateWorkloadTests.cpp | 906 unsigned int concatAxis) in NeonCreateConcatWorkloadTest() argument
912 … = CreateConcatWorkloadTest<ConcatWorkloadType, DataType>(factory, graph, outputShape, concatAxis); in NeonCreateConcatWorkloadTest()
|
| /external/armnn/src/backends/reference/test/ |
| D | RefCreateWorkloadTests.cpp | 1008 unsigned int concatAxis) in RefCreateConcatWorkloadTest() argument
1012 … = CreateConcatWorkloadTest<ConcatWorkloadType, DataType>(factory, graph, outputShape, concatAxis); in RefCreateConcatWorkloadTest()
|
| /external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
| D | armnn_descriptors.i | 228 void SetConcatAxis(unsigned int concatAxis);
|
| /external/armnn/src/backends/cl/test/ |
| D | ClCreateWorkloadTests.cpp | 1136 unsigned int concatAxis) in ClCreateConcatWorkloadTest() argument
1142 … = CreateConcatWorkloadTest<ConcatWorkloadType, DataType>(factory, graph, outputShape, concatAxis); in ClCreateConcatWorkloadTest()
|
| /external/armnn/src/armnnSerializer/ |
| D | ArmnnSchema.fbs | 762 concatAxis:uint;
|
| /external/armnn/include/armnn/ |
| D | Descriptors.hpp | 230 void SetConcatAxis(unsigned int concatAxis);
|
| /external/armnn/src/armnnTestUtils/ |
| D | CreateWorkload.hpp | 2046 unsigned int concatAxis) in CreateConcatWorkloadTest() argument
2060 concatAxis); in CreateConcatWorkloadTest()
|
| /external/armnn/generated/ |
| D | ArmnnSchema_generated.h | 7727 uint32_t concatAxis() const {
7755 void add_concatAxis(uint32_t concatAxis) {
7756 fbb_.AddElement<uint32_t>(OriginsDescriptor::VT_CONCATAXIS, concatAxis, 0);
7780 uint32_t concatAxis = 0,
7788 builder_.add_concatAxis(concatAxis);
7794 uint32_t concatAxis = 0,
7801 concatAxis,
|
| /external/armnn/src/armnnDeserializer/ |
| D | Deserializer.cpp | 2145 descriptor.SetConcatAxis(originsDescriptor->concatAxis()); in ParseConcat()
|