Searched refs:inputOutputTensorShape (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | L2NormalizationTestImpl.cpp | 29 const armnn::TensorShape& inputOutputTensorShape, in L2NormalizationTestImpl() argument
40 const armnn::TensorInfo inputTensorInfo(inputOutputTensorShape, ArmnnType, scale, offset); in L2NormalizationTestImpl()
41 … const armnn::TensorInfo outputTensorInfo(inputOutputTensorShape, ArmnnType, outScale, outOffset); in L2NormalizationTestImpl()
709 const armnn::TensorShape inputOutputTensorShape = armnn::TensorShape({ 5, 2 }); in L2Normalization2dShapeTest() local
729 … const armnn::TensorInfo inputTensorInfo(inputOutputTensorShape, armnn::DataType::Float32, 0.f, 0); in L2Normalization2dShapeTest()
730 …const armnn::TensorInfo outputTensorInfo(inputOutputTensorShape, armnn::DataType::Float32, 0.f, 0); in L2Normalization2dShapeTest()
|
| H A D | BatchNormalizationTestImpl.cpp | 34 const armnn::TensorShape& inputOutputTensorShape, in BatchNormTestImpl() argument
42 armnn::TensorInfo inputTensorInfo(inputOutputTensorShape, ArmnnType); in BatchNormTestImpl()
43 armnn::TensorInfo outputTensorInfo(inputOutputTensorShape, ArmnnType); in BatchNormTestImpl()
47 armnn::TensorInfo tensorInfo({ inputOutputTensorShape[dataLayoutIndexed.GetChannelsIndex()] }, in BatchNormTestImpl()
|