Searched defs:SubtractionTest (Results 1 – 2 of 2) sorted by relevance
225 struct SubtractionTest struct227 using LayerType = armnn::SubtractionLayer;228 static std::string GetReceiverLayerName() { return "Subtraction"; }; in GetReceiverLayerName()229 static const bool isElementWise = true;231 static TensorShape GetInputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCin in GetInputShape()232 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape()234 constexpr static const unsigned int inputSize = 48; // batchIn * heightIn * widthIn * channelIn235 …onstexpr static const unsigned int outputSize = 48; // batchOut * heightOut * widthOut * channelOut237 static IConnectableLayer* AddReceiverLayer(INetwork* network, in AddReceiverLayer()
107 LayerTestResult<float, 4> SubtractionTest( in SubtractionTest() function