Searched defs:filterShape (Results 1 – 8 of 8) sorted by relevance
| /external/armnn/delegate/src/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
179 std::vector<int32_t> filterShape { 1, 2, 2, 1 }; in Conv2DWithBiasesRelu6Uint8Test() local
278 std::vector<int32_t> filterShape { 1, 2, 2, 1 }; in TransposeConvInt8Test() local
311 std::vector<int32_t> filterShape { 1, 2, 2, 1 }; in TransposeConvFp32Test() local
|
| D | DepthwiseConvolution2dTest.cpp | 26 std::vector<int32_t> filterShape { 1, 2, 2, 4 }; in DepthwiseConv2dValidReluFp32Test() local
86 std::vector<int32_t> filterShape { 1, 3, 3, 1 }; in DepthwiseConv2dSameUint8Test() local
|
| D | ConvolutionTestHelper.hpp | 188 std::vector<int32_t>& filterShape, in ConvolutionTest()
|
| /external/armnn/src/armnn/layers/ |
| D | Convolution2dLayer.cpp | 37 const TensorShape filterShape = inputShapes[1]; in SerializeLayerParameters() local
87 const TensorShape filterShape = inputShapes[1]; in InferOutputShapes() local
|
| D | DepthwiseConvolution2dLayer.cpp | 36 const TensorShape filterShape = inputShapes[1]; in SerializeLayerParameters() local
89 const TensorShape& filterShape = inputShapes[1]; in InferOutputShapes() local
|
| /external/armnn/src/armnn/test/ |
| D | InferOutputTests.hpp | 458 armnn::TensorShape filterShape(4, filterSize.data()); in Convolution2dInferOutputShapeTest() local
489 armnn::TensorShape filterShape(4, filterSize.data()); in TransposeConvolution2dInferOutputShapeTest() local
522 armnn::TensorShape filterShape(4, filterSize.data()); in DepthwiseConvolution2dInferOutputShapeTest() local
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonLayerTests.cpp | 290 const TensorShape& filterShape = weightsInfo.GetShape(); in CreateOutputTensorInfo() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 341 auto filterShape = filter_type.getShape(); in legalizeFilter() local
|