Home
last modified time | relevance | path

Searched defs:weightsInfo (Results 1 – 25 of 31) sorted by relevance

12

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DTransposeConvolution2dTestImpl.cpp123 armnn::TensorInfo& weightsInfo, in TransposeConvolution2dTest()
209 armnn::TensorInfo& weightsInfo, in SwizzleData()
247 TensorInfo weightsInfo(weightsShape, ArmnnType); in SimpleTransposeConvolution2dTest() local
336 TensorInfo weightsInfo(weightsShape, ArmnnType); in PaddedTransposeConvolution2dTest() local
427 TensorInfo weightsInfo(weightsShape, ArmnnType); in StridedTransposeConvolution2dTest() local
508 TensorInfo weightsInfo(weightsShape, ArmnnType); in MultiChannelTransposeConvolution2dTest() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DLayerReleaseConstantDataTest.cpp89 TensorInfo weightsInfo = TensorInfo({ 2, 3, 5, 3 }, armnn::DataType::Float32, 1.0, 0.0, true); variable
138 TensorInfo weightsInfo = TensorInfo({ 3, 3, 5, 3 }, armnn::DataType::Float32, 1.0, 0.0, true); variable
185 TensorInfo weightsInfo = TensorInfo({ 7, 20 }, DataType::QAsymmU8, inputsQScale, 0.0, true); variable
H A DConvolution3dEndToEndTestImpl.hpp23 const armnn::TensorInfo& weightsInfo, in CreateConvolution3dNetwork()
61 TensorInfo weightsInfo({ 3, 3, 3, 1, 1 }, ArmnnType, qScale, qOffset, true); in Convolution3dEndToEnd() local
H A DConvolution2dEndToEndTestImpl.hpp23 const armnn::TensorInfo& weightsInfo, in CreateConstConvolution2dNetwork()
64 TensorInfo weightsInfo({ 1, 3, 3, 1 }, ArmnnType, qScale, qOffset, true); in Convolution2dEndToEnd() local
H A DDepthwiseConvolution2dEndToEndTests.hpp23 const armnn::TensorInfo& weightsInfo, in CreateDepthwiseConvolution2dNetwork()
76 …TensorInfo weightsInfo({1, kernelHeight, kernelWidth, outputChannels}, ArmnnType, qScale, qOffset,… in DepthwiseConvolution2dEndToEnd() local
H A DTransposeConvolution2dEndToEndTestImpl.hpp70 TensorInfo weightsInfo(weightsShape, ArmnnType, qScale, qOffset, true); in TransposeConvolution2dEndToEnd() local
163 TensorInfo weightsInfo({1, 2, 2, 1}, ArmnnType, qScale, qOffset, true); in SimpleTransposeConvolution2dEndToEnd() local
H A DOptimizedNetworkTests.cpp382 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 1 }, armnn::DataType::Float32, 0.0f, 0, true); variable
/aosp_15_r20/external/armnn/src/backends/tosaReference/test/
H A DTosaRefLayerSupportTests.cpp152 TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32); variable
175 TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
511 TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32); variable
535 TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
/aosp_15_r20/external/armnn/src/backends/tosaCommon/test/
H A DOneToOneMappingTests.cpp189 const TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
225 const TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
623 const TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
669 const TensorInfo weightsInfo({ 1, 3, 3, 1 }, DataType::Float32, 0.0f, 0, true); variable
/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A DConvertConstantsFloatToHalfTests.cpp28 …armnn::TensorInfo weightsInfo = armnn::TensorInfo(4, dims, armnn::DataType::Float32, 0.0f, 0, true… variable
87 const armnn::TensorInfo weightsInfo({ 4 }, armnn::DataType::Float32, 0.0f, 0, true); variable
H A DFoldPadTests.cpp28 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 1.0f, 0, true); variable
110 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 1.0f, 0, true); variable
691 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 1.0f, 0, true); variable
815 TensorInfo weightsInfo(4, weightsShape, DataType::Float32, 0.0f, 0, true); variable
H A DFuseActivationTests.cpp79 TensorInfo weightsInfo(GetWeightsShape(), ArmnnType, scale, offset, true); in AddConstantLayers() local
131 TensorInfo weightsInfo(GetWeightsShape(), ArmnnType, scale, offset, true); in AddConstantLayers() local
179 TensorInfo weightsInfo(GetWeightsShape(), ArmnnType, scale, offset, true); in AddConstantLayers() local
H A DFuseBatchNormTests.cpp145 TensorInfo weightsInfo(4, weightsDimensionSizes, ArmnnType, 0.0f, 0, true); in CreateNetwork() local
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DSerializerTests.cpp421 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 1 }, armnn::DataType::Float32, 0.0f, 0, true); variable
528 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 1 }, armnn::DataType::Float32, 0.0f, 0, true); variable
583 const armnn::TensorInfo weightsInfo({ 3, 3, 3, 1, 1 }, armnn::DataType::Float32, 0.0f, 0, true); variable
668 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 3 }, armnn::DataType::Float32, 0.0f, 0, true); variable
786 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 1 }, armnn::DataType::Float32, 0.0f, 0, true); variable
1159 const armnn::TensorInfo weightsInfo({ 5, 3 }, armnn::DataType::Float32, 0.0f, 0, true); variable
1203 const armnn::TensorInfo weightsInfo({ 5, 3 }, armnn::DataType::Float32); variable
1248 const armnn::TensorInfo weightsInfo({ 5, 3 }, armnn::DataType::Float32, 0.0f, 0, true); variable
2869 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 1 }, armnn::DataType::Float32, 0.0f, 0, true); variable
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadUtils.cpp144 TensorInfo weightsInfo = weightTensor->GetTensorInfo(); in Convert1HWOTensorToAcl() local
206 TensorInfo weightsInfo = weightTensor->GetTensorInfo(); in Convert1HWOtoMIHW() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefTransposeConvolution2dWorkload.cpp22 const TensorInfo& weightsInfo = m_Weights->GetTensorInfo(); in RefTransposeConvolution2dWorkload() local
/aosp_15_r20/external/armnn/samples/
H A DSimpleSample.cpp31 TensorInfo weightsInfo(TensorShape({1, 1}), DataType::Float32, 0.0f, 0, true); in main() local
H A DAsyncExecutionSample.cpp52 TensorInfo weightsInfo(TensorShape({1, 1}), DataType::Float32, 0.0f, 0, true); in main() local
H A DCustomMemoryAllocatorSample.cpp81 TensorInfo weightsInfo(TensorShape({1, 1}), DataType::Float32, 0.0f, 0, true); in main() local
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonDepthwiseConvolutionWorkload.cpp103 arm_compute::ITensorInfo* weightsInfo = weights.info(); in NeonDepthwiseConvolutionWorkload() local
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClDepthwiseConvolutionWorkload.cpp116 arm_compute::ITensorInfo* weightsInfo = weights.info(); in ClDepthwiseConvolutionWorkload() local
/aosp_15_r20/external/armnn/src/backends/neon/test/
H A DNeonLayerTests.cpp384 const TensorInfo& weightsInfo, in CreateOutputTensorInfo()
/aosp_15_r20/external/armnn/src/backends/cl/test/
H A DClCustomAllocatorTests.cpp68 TensorInfo weightsInfo(TensorShape({1, 1}), DataType::Float32, 0.0f, 0, true); in CreateTestNetwork() local
/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils_1_2.hpp439 armnn::TensorInfo weightsInfo = weightsInput.GetTensorInfo(); in ConvertConv2d_1_2() local
612 armnn::TensorInfo weightsInfo = weightsInput.GetTensorInfo(); in ConvertDepthwiseConv2d_1_2() local
1179 const TensorInfo& weightsInfo = weights.GetInfo(); in ConvertGroupedConv2d() local
/aosp_15_r20/external/armnn/src/armnn/test/
H A DGraphTests.cpp632 TensorInfo weightsInfo = constInfo; variable

12