Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/delegate/test/
H A DConvolution2dTest.cpp26 std::vector<int32_t> filterShape { 1, 3, 3, 1 }; in Conv2DWithBiasesFp32Test() local
80 std::vector<int32_t> filterShape { 1, 2, 2, 1 }; in Conv2DWithBiasesInt8Test() local
123 std::vector<int32_t> filterShape { 1, 2, 2, 1 }; in Conv2DWithBiasesReluUint8Test() local
181 std::vector<int32_t> filterShape { 1, 2, 2, 1 }; in Conv2DWithBiasesRelu6Uint8Test() local
230 std::vector<int32_t> filterShape { 4,2,2,2 }; in Conv2DPerChannelInt8Test() local
375 std::vector<int32_t> filterShape { 1, 2, 2, 1 }; in TransposeConvInt8Test() local
408 std::vector<int32_t> filterShape { 1, 2, 2, 1 }; in TransposeConvFp32Test() local
H A DConvolution3dTest.cpp42 std::vector<int32_t> filterShape { 2, 2, 2, 1, 1 }; in Conv3DWithBiasesSimpleWithPaddingFp32Test() local
83 std::vector<int32_t> filterShape { 3, 5, 5, 1, 1 }; in Conv3DWithBiasesStridesFp32Test() local
142 std::vector<int32_t> filterShape { 2, 2, 2, 2, 2 }; in Conv3DWithBiasesDilationFp32Test() local
193 std::vector<int32_t> filterShape { 3, 3, 3, 1, 1 }; in Conv3DFp32SmallTest() local
H A DDepthwiseConvolution2dTest.cpp26 std::vector<int32_t> filterShape { 1, 2, 2, 4 }; in DepthwiseConv2dValidReluFp32Test() local
88 std::vector<int32_t> filterShape { 1, 3, 3, 1 }; in DepthwiseConv2dSameUint8Test() local
135 std::vector<int32_t> filterShape { 1, 2, 2, 16 }; in DepthwiseConv2dSameInt8PerChannelTest() local
H A DConvolutionTestHelper.hpp207 std::vector<int32_t>& filterShape, in ConvolutionTest()
443 std::vector<int32_t>& filterShape, in Convolution3dTest()
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DConv2D.cpp106 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
175 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData()
192 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc()
221 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
275 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
304 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc()
331 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in conv()
350 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc()
443 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc()
507 const Shape& filterShape, const float* filterScales, in convQuant8PerChannel()
DDepthwiseConv2D.cpp93 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
130 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc()
164 const _Float16* filterData, const Shape& filterShape, in depthwiseConvNhwc()
190 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
241 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
274 const Shape& filterShape, const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannelNhwc()
370 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in depthwiseConv()
391 const int8_t* filterData, const Shape& filterShape, in depthwiseConvQuant8PerChannel()
DTransposeConv2D.cpp70 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
123 const Shape& filterShape, const float* biasData, const Shape& /*biasShape*/, in transposeConvNhwc()
180 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc()
274 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc()
298 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in transposeConv()
314 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannelNhwc()
417 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannel()
DGroupedConv2D.cpp51 const Shape& filterShape, const float* biasData, const Shape& /*biasShape*/, in groupedConvFloat32()
109 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8()
205 const int8_t* filterData, const Shape& filterShape, in groupedConvQuant8PerChannel()
291 const _Float16* filterData, const Shape& filterShape, in groupedConvFloat16()
/aosp_15_r20/external/armnn/src/armnn/test/
H A DInferOutputTests.hpp458 armnn::TensorShape filterShape(4, filterSize.data()); in Convolution2dInferOutputShapeTest() local
495 armnn::TensorShape filterShape(5, filterSize.data()); in Convolution3dInferOutputShapeTest() local
526 armnn::TensorShape filterShape(4, filterSize.data()); in TransposeConvolution2dInferOutputShapeTest() local
559 armnn::TensorShape filterShape(4, filterSize.data()); in DepthwiseConvolution2dInferOutputShapeTest() local
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DDepthwiseConvolution2dLayer.cpp36 const TensorShape filterShape = inputShapes[1]; in SerializeLayerParameters() local
69 const TensorShape& filterShape = inputShapes[1]; in InferOutputShapes() local
H A DConvolution2dLayer.cpp37 const TensorShape filterShape = inputShapes[1]; in SerializeLayerParameters() local
68 const TensorShape filterShape = inputShapes[1]; in InferOutputShapes() local
H A DConvolution3dLayer.cpp32 const TensorShape filterShape = inputShapes[1]; in SerializeLayerParameters() local
66 const TensorShape& filterShape = inputShapes[1]; in InferOutputShapes() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefDepthwiseConvolution2dWorkload.cpp57 const TensorShape& filterShape = GetTensorInfo(inputs[1]).GetShape(); in Execute() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DOperationsExecutionUtils.cpp154 bool GetQuantizedConvolutionMultiplier(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultiplier()
169 bool GetQuantizedConvolutionMultiplier(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultiplier()
DCpuExecutor.cpp1511 Shape filterShape = filter.shape(); in executeOperation() local
/aosp_15_r20/external/armnn/src/backends/neon/test/
H A DNeonLayerTests.cpp389 const TensorShape& filterShape = weightsInfo.GetShape(); in CreateOutputTensorInfo() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dprepare_tf.cc383 auto filterShape = filter_type.getShape(); in legalizeFilter() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_to_linalg.cc2430 RankedTensorType filterShape = in matchAndRewrite() local