Home
last modified time | relevance | path

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

/external/armnn/src/backends/backendsCommon/test/layerTests/
DTransposeConvolution2dTestImpl.cpp123 armnn::TensorInfo& weightsInfo, in TransposeConvolution2dTest()
210 armnn::TensorInfo& weightsInfo, in SwizzleData()
248 TensorInfo weightsInfo(weightsShape, ArmnnType); in SimpleTransposeConvolution2dTest() local
337 TensorInfo weightsInfo(weightsShape, ArmnnType); in PaddedTransposeConvolution2dTest() local
428 TensorInfo weightsInfo(weightsShape, ArmnnType); in StridedTransposeConvolution2dTest() local
509 TensorInfo weightsInfo(weightsShape, ArmnnType); in MultiChannelTransposeConvolution2dTest() local
/external/armnn/src/backends/reference/workloads/
DRefTransposeConvolution2dWorkload.cpp22 const TensorInfo& weightsInfo = m_Weights->GetTensorInfo(); in RefTransposeConvolution2dWorkload() local
/external/armnn/samples/
DSimpleSample.cpp32 TensorInfo weightsInfo(TensorShape({1, 1}), DataType::Float32); in main() local
/external/armnn/src/backends/neon/test/
DNeonLayerTests.cpp285 const TensorInfo& weightsInfo, in CreateOutputTensorInfo()
/external/armnn/src/backends/backendsCommon/test/
DTransposeConvolution2dEndToEndTestImpl.hpp75 TensorInfo weightsInfo(weightsShape, ArmnnType, qScale, qOffset); in TransposeConvolution2dEndToEnd() local
/external/armnn/src/armnn/optimizations/
DFuseBatchNorm.hpp54 auto weightsInfo(convLayer->m_Weight->GetTensorInfo()); in Run() local
/external/armnn/src/armnn/test/optimizations/
DFuseBatchNormTests.cpp110 TensorInfo weightsInfo(4, weightsDimensionSizes, ArmnnType); in CreatNetwork() local
DFuseActivationTests.cpp71 TensorInfo weightsInfo(GetWeightsShape(), ArmnnType, g_qScale, g_qOffset); in AddReceiverLayer() local
108 TensorInfo weightsInfo(GetWeightsShape(), ArmnnType, g_qScale, g_qOffset); in AddReceiverLayer() local
141 TensorInfo weightsInfo(GetWeightsShape(), ArmnnType, g_qScale, g_qOffset); in AddReceiverLayer() local
/external/armnn/src/armnnSerializer/test/
DSerializerTests.cpp585 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 1 }, armnn::DataType::Float32); in BOOST_AUTO_TEST_CASE() local
800 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 3 }, armnn::DataType::Float32); in BOOST_AUTO_TEST_CASE() local
1301 const armnn::TensorInfo weightsInfo({ 5, 3 }, armnn::DataType::Float32); in BOOST_AUTO_TEST_CASE() local
2910 const armnn::TensorInfo weightsInfo({ 1, 3, 3, 1 }, armnn::DataType::Float32); in BOOST_AUTO_TEST_CASE() local
/external/android-nn-driver/
DConversionUtils_1_2.hpp984 const TensorInfo& weightsInfo = weights.GetInfo(); in ConvertGroupedConv2d() local