Searched defs:weightsTensorInfo (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
H A D | FullyConnectedEndToEndTestImpl.hpp | 24 … const armnn::TensorInfo& weightsTensorInfo, in CreateFullyConnectedNetworkNonConstWeights() 43 … const armnn::TensorInfo& weightsTensorInfo, in CreateFullyConnectedNetworkNonConstWeightsConstBias() 66 … const armnn::TensorInfo& weightsTensorInfo, in CreateFullyConnectedNetworkConstWeightsNonConstBias() 146 … const armnn::TensorInfo& weightsTensorInfo, in CreateFullyConnectedNetworkNoConnectedBiasExplicit() 178 armnn::TensorInfo weightsTensorInfo({ 2, 6 }, ArmnnType); in FullyConnectedWithDynamicWeightsEndToEnd() local
|
/aosp_15_r20/external/armnn/src/armnn/test/optimizations/ |
H A D | PermuteDepthwiseConv2dWeightsTests.cpp | 20 const armnn::TensorInfo& weightsTensorInfo, in CreateSimpleDepthwiseConv2dNetwork() 81 armnn::TensorInfo weightsTensorInfo({ 2, 6 }, armnn::DataType::Float32); in PermuteDepthwiseConv2dWeightsTest() local
|
/aosp_15_r20/external/armnn/src/armnnTestUtils/ |
H A D | CreateWorkload.hpp | 533 armnn::TensorInfo weightsTensorInfo(weightShape, DataType, inputsQScale); local 597 armnn::TensorInfo weightsTensorInfo(weightShape, DataType, inputsQScale); local 691 armnn::TensorInfo weightsTensorInfo(weightShape, DataType, inputsQScale); local 1149 armnn::TensorInfo weightsTensorInfo(weightShape, DataType, inputsQScale); in CreateDirectConvolution2dWorkloadTest() local 1263 armnn::TensorInfo weightsTensorInfo({7, 20}, DataType, inputsQScale); in CreateFullyConnectedWorkloadTest() local 1308 armnn::TensorInfo weightsTensorInfo({7, 20}, DataType, inputsQScale); in CreateFullyConnectedWithBlobWorkloadTest() local 2104 TensorInfo weightsTensorInfo(TensorShape({16, 1, 1, 16}), dataType, 0.9f, 0, true); in CreatePreCompiledWorkloadTest() local
|
/aosp_15_r20/external/armnn/delegate/classic/src/ |
H A D | FullyConnected.hpp | 58 const armnn::TensorInfo& weightsTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteWeightsTensor); in VisitFullyConnectedOperator() local
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
H A D | FullyConnectedTestImpl.cpp | 30 armnn::TensorInfo weightsTensorInfo, in SimpleFullyConnectedTestImpl()
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/ |
H A D | WorkloadData.cpp | 183 const TensorInfo& weightsTensorInfo, in ValidateBiasTensorQuantization()
|