Searched defs:Convolution2dTest (Results 1 – 1 of 1) sorted by relevance
| /external/armnn/src/armnn/test/optimizations/ |
| D | FuseActivationTests.cpp | 40 struct Convolution2dTest struct
42 using LayerType = Convolution2dLayer;
43 static const bool isElementWise = false;
44 static const bool isConstTensorAsInputSupported = true;
46 static TensorShape GetInputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCin in GetInputShape()
47 static TensorShape GetOutputShape() { return TensorShape( {1, 3, 3, 4}); } // NHWCout in GetOutputShape()
48 static TensorShape GetWeightsShape() { return TensorShape( {4, 2, 2, 3}); } // CoutHWCin in GetWeightsShape()
50 constexpr static const unsigned int inputSize = 48; // batchIn * heightIn * widthIn * channelIn
51 …onstexpr static const unsigned int outputSize = 36; // batchOut * heightOut * widthOut * channelOut
53 static IConnectableLayer* AddReceiverLayer(INetwork* network, in AddReceiverLayer()
[all …]
|