| /external/armnn/src/armnnDeserializer/test/ |
| D | DeserializeConvolution2d.cpp | 17 const std::string & weightsShape, in Convolution2dFixture()
81 dimensions: )" + weightsShape + R"(, in Convolution2dFixture()
128 const std::string & weightsShape, in Convolution2dFixtureConstantAsInput()
217 dimensions: )" + weightsShape + R"(, in Convolution2dFixtureConstantAsInput()
232 dimensions: )" + weightsShape + R"(, in Convolution2dFixtureConstantAsInput()
|
| D | DeserializeConvolution3d.cpp | 17 const std::string& weightsShape, in Convolution3dFixture()
69 dimensions: )" + weightsShape + R"(, in Convolution3dFixture()
85 dimensions: )" + weightsShape + R"(, in Convolution3dFixture()
|
| D | DeserializeFullyConnected.cpp | 17 const std::string & weightsShape, in FullyConnectedFixture()
77 dimensions: )" + weightsShape + R"(, in FullyConnectedFixture()
|
| /external/android-nn-driver/1.0/ |
| D | FullyConnected.hpp | 16 const armnn::TensorShape& weightsShape) in FlattenFullyConnectedInput() argument
21 unsigned int inputSize = weightsShape[1]; in FlattenFullyConnectedInput()
39 const armnn::TensorShape& weightsShape, in VerifyFullyConnectedShapes() argument
44 return (inputShape[0] == outputShape[0] && weightsShape[dimIdx] == outputShape[1]); in VerifyFullyConnectedShapes()
|
| /external/armnn/src/armnn/test/optimizations/ |
| D | PermuteDepthwiseConv2dWeightsTests.cpp | 66 unsigned int weightsShape[] = {outputShape[0], outputShape[1], outputShape[2], outputShape[3]}; in PermuteDepthwiseConv2dWeightsTestRunner() local
71 CHECK((newShape[0] == weightsShape[0])); in PermuteDepthwiseConv2dWeightsTestRunner()
72 CHECK((newShape[1] == weightsShape[3])); in PermuteDepthwiseConv2dWeightsTestRunner()
73 CHECK((newShape[2] == weightsShape[1])); in PermuteDepthwiseConv2dWeightsTestRunner()
74 CHECK((newShape[3] == weightsShape[2])); in PermuteDepthwiseConv2dWeightsTestRunner()
|
| D | FoldPadTests.cpp | 23 const unsigned int weightsShape[] = {1, 2, 3, 3}; variable
28 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 1.0f, 0, true);
105 const unsigned int weightsShape[] = {1, 2, 3, 3}; variable
110 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 1.0f, 0, true);
646 const unsigned int weightsShape[] = {4, 2, 2, 3}; // CoutHWCin variable
691 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 1.0f, 0, true);
770 const unsigned int weightsShape[] = {1, 2, 2, 12}; // 1HWCout variable
815 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 0.0f, 0, true);
|
| /external/armnn/src/armnn/optimizations/ |
| D | FuseBatchNorm.hpp | 65 auto weightsShape = weightsTensor.GetInfo().GetShape(); in Run() local
67 const unsigned int depthMultiplier = depthwise ? weightsShape[3] / inputChannels : 1; in Run()
68 const unsigned int outputChannels = depthwise ? weightsShape[3] : weightsShape[0]; in Run()
69 const unsigned int weightsHeight = depthwise ? weightsShape[1] : in Run()
70 weightsShape[dataLayout.GetHeightIndex()]; in Run()
71 const unsigned int weightsWidth = depthwise ? weightsShape[2] : in Run()
72 weightsShape[dataLayout.GetWidthIndex()]; in Run()
|
| /external/armnn/src/backends/reference/workloads/ |
| D | TransposeConvolution2d.cpp | 20 const TensorShape& weightsShape, in TransposeConvolution2dImpl() argument
39 const unsigned int weightsHeight = weightsShape[heightIndex]; in TransposeConvolution2dImpl()
40 const unsigned int weightsWidth = weightsShape[widthIndex]; in TransposeConvolution2dImpl()
41 const unsigned int weightsDepth = weightsShape[channelsIndex]; in TransposeConvolution2dImpl()
56 const std::vector<float> filterVec = weightsDecoder.DecodeTensor(weightsShape); in TransposeConvolution2dImpl()
|
| D | TransposeConvolution2d.hpp | 22 const TensorShape& weightsShape,
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | FullyConnectedTestImpl.cpp | 194 unsigned int weightsShape[] = { inputChannels, outputChannels }; in FullyConnectedLargeTestCommon() local
197 std::swap(weightsShape[0], weightsShape[1]); in FullyConnectedLargeTestCommon()
204 weightsDesc = armnn::TensorInfo(2, weightsShape, ArmnnType); in FullyConnectedLargeTestCommon()
294 unsigned int weightsShape[] = { inputChannels, outputChannels }; in FullyConnectedFloat32Test() local
298 std::swap(weightsShape[0], weightsShape[1]); in FullyConnectedFloat32Test()
305 weightsDesc = armnn::TensorInfo(2, weightsShape, armnn::DataType::Float32); in FullyConnectedFloat32Test()
|
| D | TransposeConvolution2dTestImpl.cpp | 243 TensorShape weightsShape = { batches, channels, hWeights, wWeights }; in SimpleTransposeConvolution2dTest() local
247 TensorInfo weightsInfo(weightsShape, ArmnnType); in SimpleTransposeConvolution2dTest()
332 TensorShape weightsShape = { batches, channels, hWeights, wWeights }; in PaddedTransposeConvolution2dTest() local
336 TensorInfo weightsInfo(weightsShape, ArmnnType); in PaddedTransposeConvolution2dTest()
423 TensorShape weightsShape = { batches, channels, hWeights, wWeights }; in StridedTransposeConvolution2dTest() local
427 TensorInfo weightsInfo(weightsShape, ArmnnType); in StridedTransposeConvolution2dTest()
503 TensorShape weightsShape = { 2, 1, 3, 3 }; in MultiChannelTransposeConvolution2dTest() local
508 TensorInfo weightsInfo(weightsShape, ArmnnType); in MultiChannelTransposeConvolution2dTest()
|
| /external/armnn/src/armnn/test/ |
| D | OptimizerTests.cpp | 424 const unsigned int* weightsShape, const unsigned int* outputShape, in CreateConvolution2dGraph() argument
432 armnn::TensorInfo(4, weightsShape, armnn::DataType::Float32, 0.0f, 0, true), in CreateConvolution2dGraph()
462 const unsigned int weightsShape[] = { 2, 3, 5, 3 }; variable
464 CreateConvolution2dGraph(graph, inputShape, weightsShape, outputShape);
473 const unsigned int weightsShape[] = { 2, 5, 3, 3 }; variable
475 CreateConvolution2dGraph(graph, inputShape, weightsShape, outputShape, DataLayout::NHWC);
481 … const unsigned int* weightsShape, const unsigned int* outputShape, in CreateDepthwiseConvolution2dGraph() argument
486 …armnn::TensorInfo weightsInfo(TensorShape(4, weightsShape), armnn::DataType::Float32, 0.0f, 0, tru… in CreateDepthwiseConvolution2dGraph()
517 const unsigned int weightsShape[] = { 1, 3, 3, 2 }; variable
519 CreateDepthwiseConvolution2dGraph(graph, inputShape, weightsShape, outputShape);
[all …]
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | FullyConnectedEndToEndTestImpl.hpp | 235 unsigned int weightsShape[] = { inputChannels, outputChannels }; in FullyConnectedWithDynamicOrConstantInputsEndToEnd() local
239 std::swap(weightsShape[0], weightsShape[1]); in FullyConnectedWithDynamicOrConstantInputsEndToEnd()
246 …armnn::TensorInfo weightsDesc = armnn::TensorInfo(2, weightsShape, armnn::DataType::Float32, 0.0f,… in FullyConnectedWithDynamicOrConstantInputsEndToEnd()
356 unsigned int weightsShape[] = { inputChannels, outputChannels }; in FullyConnectedErrorChecking() local
362 …armnn::TensorInfo weightsDesc = armnn::TensorInfo(2, weightsShape, armnn::DataType::Float32, 0.0f,… in FullyConnectedErrorChecking()
|
| D | TransposeConvolution2dEndToEndTestImpl.hpp | 63 TensorShape weightsShape = MakeTensorShape(batches, channels, hWeights, wWeights, dataLayout); in TransposeConvolution2dEndToEnd() local
70 TensorInfo weightsInfo(weightsShape, ArmnnType, qScale, qOffset, true); in TransposeConvolution2dEndToEnd()
|
| D | WorkloadDataValidation.cpp | 150 unsigned int weightsShape[] = { 1, 1, inputChannels, outputChannels }; variable
155 weightsDesc = armnn::TensorInfo(4, weightsShape, armnn::DataType::Float32);
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadUtils.cpp | 215 auto weightsShape = weightsInfo.GetShape(); in Convert1HWOtoMIHW() local
217 unsigned int depthMultiplier = weightsShape[3] / inputInfo.GetShape()[channelIndex]; in Convert1HWOtoMIHW()
218 weightsInfo.SetShape({ weightsShape[1], in Convert1HWOtoMIHW()
219 weightsShape[2], in Convert1HWOtoMIHW()
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonDepthwiseConvolutionWorkload.cpp | 105 auto weightsShape = weightsInfo->tensor_shape(); in NeonDepthwiseConvolutionWorkload() local
111 … ComputeDepthwiseConv2dDepthMultiplier(m_Data.m_Parameters.m_DataLayout, weightsShape, inputShape); in NeonDepthwiseConvolutionWorkload()
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClDepthwiseConvolutionWorkload.cpp | 118 auto weightsShape = weightsInfo->tensor_shape(); in ClDepthwiseConvolutionWorkload() local
124 … ComputeDepthwiseConv2dDepthMultiplier(m_Data.m_Parameters.m_DataLayout, weightsShape, inputShape); in ClDepthwiseConvolutionWorkload()
|
| /external/armnn/src/backends/aclCommon/ |
| D | ArmComputeTensorUtils.cpp | 443 const arm_compute::TensorShape& weightsShape, in ComputeDepthwiseConv2dDepthMultiplier() argument
449 … depthMultiplier = static_cast<uint32_t>(weightsShape[0]) / static_cast<uint32_t>(inputShape[0]); in ComputeDepthwiseConv2dDepthMultiplier()
453 … depthMultiplier = static_cast<uint32_t>(weightsShape[2]) / static_cast<uint32_t>(inputShape[2]); in ComputeDepthwiseConv2dDepthMultiplier()
|
| D | ArmComputeTensorUtils.hpp | 108 const arm_compute::TensorShape& weightsShape,
|
| /external/armnn/python/pyarmnn/test/ |
| D | test_network.py | 282 weightsShape = ( inputChannels, outputChannels )
315 w_info = ann.TensorInfo(ann.TensorShape(weightsShape), ann.DataType_Float32, 0, 0, True)
|
| /external/armnn/shim/sl/canonical/ |
| D | ConversionUtils.hpp | 247 const armnn::TensorShape& weightsShape) in FlattenFullyConnectedInput() argument
252 unsigned int inputSize = weightsShape[1]; in FlattenFullyConnectedInput()
270 const armnn::TensorShape& weightsShape, in VerifyFullyConnectedShapes() argument
275 return (inputShape[0] == outputShape[0] && weightsShape[dimIdx] == outputShape[1]); in VerifyFullyConnectedShapes()
|
| D | Converter.cpp | 2185 const TensorShape& weightsShape = weights.GetShape(); in ConvertGroupedConv2d() local
2230 const uint32_t kernelX = weightsShape[widthIndex]; in ConvertGroupedConv2d()
2231 const uint32_t kernelY = weightsShape[heightIndex]; in ConvertGroupedConv2d()
2242 const unsigned int outputChannels = weightsShape[0]; in ConvertGroupedConv2d()
2244 const unsigned int channelsPerGroup = weightsShape[channelsIndex]; in ConvertGroupedConv2d()
2320 TensorShape groupWeightsShape(weightsShape); in ConvertGroupedConv2d()
2455 concatDescriptor = ConcatDescriptor(weightsShape[0]); in ConvertGroupedConv2d()
|
| /external/android-nn-driver/ |
| D | ConversionUtils_1_2.hpp | 1036 const TensorShape& weightsShape = weights.GetShape(); in ConvertGroupedConv2d() local
1080 const uint32_t kernelX = weightsShape[widthIndex]; in ConvertGroupedConv2d()
1081 const uint32_t kernelY = weightsShape[heightIndex]; in ConvertGroupedConv2d()
1092 const unsigned int outputChannels = weightsShape[0]; in ConvertGroupedConv2d()
1094 const unsigned int channelsPerGroup = weightsShape[channelsIndex]; in ConvertGroupedConv2d()
1170 TensorShape groupWeightsShape(weightsShape); in ConvertGroupedConv2d()
1304 concatDescriptor = ConcatDescriptor(weightsShape[0]); in ConvertGroupedConv2d()
|
| /external/armnn/src/armnnDeserializer/ |
| D | Deserializer.cpp | 1455 auto weightsShape = weightsInfo.GetShape(); in ParseConstant() local
1457 weightsShape[0], in ParseConstant()
1458 weightsShape[1], in ParseConstant()
1459 weightsShape[2]*weightsShape[3]}); in ParseConstant()
1688 auto weightsShape = weightsInfo.GetShape(); in ParseDepthwiseConvolution2d() local
1690 weightsShape[0], in ParseDepthwiseConvolution2d()
1691 weightsShape[1], in ParseDepthwiseConvolution2d()
1692 weightsShape[2]*weightsShape[3]}); in ParseDepthwiseConvolution2d()
|