Searched defs:DWConvolution2dTest (Results 1 – 1 of 1) sorted by relevance
| /external/armnn/src/armnn/test/optimizations/ |
| D | FuseActivationTests.cpp | 91 struct DWConvolution2dTest struct
94 using LayerType = DepthwiseConvolution2dLayer;
95 static const bool isElementWise = false;
96 static const bool isConstTensorAsInputSupported = true;
98 static TensorShape GetInputShape() { return TensorShape( {1, 4, 4, 3}); } // [N,H,W,Cin] in GetInputShape()
99 static TensorShape GetOutputShape() { return TensorShape( {1, 3, 3, 12}); } // [N,H,W,Cout] in GetOutputShape()
100 static TensorShape GetWeightsShape() { return TensorShape( {1, 2, 2, 12}); } // [1,H,W,Cout] in GetWeightsShape()
102 constexpr static const unsigned int inputSize = 48; //batchIn * heightIn * widthIn * channelIn;
103 …nstexpr static const unsigned int outputSize = 108; //batchOut * heightOut * widthOut * channelOut;
105 static IConnectableLayer* AddReceiverLayer(INetwork* network, in AddReceiverLayer()
[all …]
|