| /external/armnn/delegate/test/ |
| D | Convolution2dTest.cpp | 26 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
|
| D | Convolution3dTest.cpp | 42 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
|
| D | DepthwiseConvolution2dTest.cpp | 26 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
|
| D | ConvolutionTestHelper.hpp | 207 std::vector<int32_t>& filterShape, in ConvolutionTest()
443 std::vector<int32_t>& filterShape, in Convolution3dTest()
|
| /external/armnn/src/armnn/layers/ |
| D | DepthwiseConvolution2dLayer.cpp | 36 const TensorShape filterShape = inputShapes[1]; in SerializeLayerParameters() local
69 const TensorShape& filterShape = inputShapes[1]; in InferOutputShapes() local
|
| D | Convolution2dLayer.cpp | 37 const TensorShape filterShape = inputShapes[1]; in SerializeLayerParameters() local
68 const TensorShape filterShape = inputShapes[1]; in InferOutputShapes() local
|
| D | Convolution3dLayer.cpp | 32 const TensorShape filterShape = inputShapes[1]; in SerializeLayerParameters() local
66 const TensorShape& filterShape = inputShapes[1]; in InferOutputShapes() local
|
| /external/armnn/src/backends/reference/workloads/ |
| D | RefDepthwiseConvolution2dWorkload.cpp | 57 const TensorShape& filterShape = GetTensorInfo(inputs[1]).GetShape(); in Execute() local
|
| /external/armnn/src/armnn/test/ |
| D | InferOutputTests.hpp | 458 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
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonLayerTests.cpp | 389 const TensorShape& filterShape = weightsInfo.GetShape(); in CreateOutputTensorInfo() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 383 auto filterShape = filter_type.getShape(); in legalizeFilter() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 2430 RankedTensorType filterShape = in matchAndRewrite() local
|