Home
last modified time | relevance | path

Searched defs:weightInfo (Results 1 – 18 of 18) sorted by relevance

/external/armnn/src/backends/backendsCommon/
DWorkloadUtils.cpp41 void ReshapeWeightsForAcl(TensorInfo& weightInfo, DataLayout dataLayout) in ReshapeWeightsForAcl()
115 TensorInfo ConvertWeightTensorInfoFromArmnnToAcl(const TensorInfo& weightInfo, DataLayout dataLayou… in ConvertWeightTensorInfoFromArmnnToAcl()
170 std::tuple<TensorInfo, unsigned int> Convert1HWOTensorInfoToAcl(const TensorInfo& weightInfo, in Convert1HWOTensorInfoToAcl()
DWorkloadData.cpp332 const TensorInfo& weightInfo, in ValidateWeightDataType()
380 const TensorInfo& weightInfo, in ValidatePerAxisQuantization()
/external/armnn/src/armnn/test/optimizations/
DConvertConstantsHalfToFloatTests.cpp28 …armnn::TensorInfo weightInfo = armnn::TensorInfo(4, dims, armnn::DataType::Float16, 0.0f, 0, true); variable
/external/armnn/src/armnn/optimizations/
DPermuteDepthwiseConv2dWeights.hpp46 TensorInfo weightInfo = layer->GetInputSlot(1).GetConnectedOutputSlot()->GetTensorInfo(); in AddPermuteLayer() local
/external/armnn/src/backends/cl/workloads/
DClTransposeConvolution2dWorkload.cpp81 const TensorInfo& weightInfo = m_Data.m_Weight->GetTensorInfo(); in ClTransposeConvolution2dWorkload() local
/external/armnn/src/backends/backendsCommon/test/
DOptimizeSubgraphViewTests.cpp314 TensorInfo weightInfo({ 16, 1, 1, 16 }, DataType::QAsymmU8, 0.9f, 0); in BuildFullyOptimizableSubgraph1() local
361 TensorInfo weightInfo({ 16, 1, 1, 16 }, DataType::QAsymmU8, 0.9f, 0); in BuildFullyOptimizableSubgraph2() local
469 TensorInfo weightInfo({ 16, 1, 1, 16 }, DataType::QAsymmU8, 0.9f, 0); in BuildPartiallySupportedSubgraph() local
559 TensorInfo weightInfo({ 16, 1, 1, 16 }, DataType::QAsymmU8, 0.9f, 0); in BuildFullyUnoptimizableSubgraph1() local
607 TensorInfo weightInfo({ 16, 1, 1, 16 }, DataType::QAsymmU8, 0.9f, 0); in BuildPartiallyOptimizableSubgraph1() local
712 TensorInfo weightInfo({ 16, 1, 1, 16 }, DataType::QAsymmU8, 0.9f, 0); in BuildPartiallyOptimizableSubgraph2() local
DLayerReleaseConstantDataTest.cpp35 armnn::TensorInfo weightInfo({3}, armnn::DataType::Float32); variable
DWorkloadDataValidation.cpp666 const TensorInfo weightInfo(weightShape, weightType, weightPerAxisScales, 0); variable
711 const TensorInfo weightInfo(weightShape, weightType, weightPerAxisScales, 0); variable
757 const TensorInfo weightInfo(weightShape, weightType, weightPerAxisScales, 0); variable
DDynamicBackendTests.hpp1451 TensorInfo weightInfo(weightShape, DataType::Float32); in CreateReferenceDynamicBackendTestImpl() local
1533 TensorInfo weightInfo(weightShape, DataType::Float32); in CreateSampleDynamicBackendTestImpl() local
/external/android-nn-driver/
DConversionUtils.cpp67 armnn::TensorInfo weightInfo = weight.GetTensorInfo(); in SanitizeQuantizationScale() local
DConversionUtils.hpp481 const armnn::TensorInfo& weightInfo, in SanitizeBiasQuantizationScale()
/external/armnn/src/profiling/test/
DProfilingTestUtils.cpp395 TensorInfo weightInfo({ 3, 2, 3, 1 }, DataType::Float32, 0.0f, 0, true); in VerifyPostOptimisationStructureTestImpl() local
/external/armnn/shim/sl/canonical/
DConversionUtils.cpp59 armnn::TensorInfo weightInfo = weight.GetTensorInfo(); in SanitizeQuantizationScale() local
DConversionUtils.hpp420 const armnn::TensorInfo& weightInfo, in SanitizeBiasQuantizationScale()
/external/armnn/src/armnnTestUtils/
DCreateWorkload.hpp390 armnn::TensorInfo weightInfo({3}, DataType); in CreateBatchNormalizationWorkloadTest() local
448 armnn::TensorInfo weightInfo({3}, DataType); in CreateBatchNormalizationWithBlobWorkloadTest() local
/external/armnn/src/armnnOnnxParser/
DOnnxParser.cpp1164 TensorInfo weightInfo = *m_TensorsInfo[weightName].m_info; in AddFullyConnected() local
2037 TensorInfo weightInfo = *m_TensorsInfo[node.input(1)].m_info; in ParseGemm() local
/external/armnn/src/backends/cl/
DClLayerSupport.cpp96 bool IsMatchingSize2d(const TensorInfo& weightInfo) in IsMatchingSize2d()
/external/armnn/src/armnn/test/
DSubgraphViewTests.cpp1904 TensorInfo weightInfo = constInfo; in ReplaceConstantMultiplicationWithDepthwise() local