Home
last modified time | relevance | path

Searched defs:SubtractionTest (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A DFuseActivationTests.cpp315 struct SubtractionTest struct
317 using LayerType = SubtractionLayer;
318 static const bool isElementWise = true;
319 static const bool isConstTensorAsInputSupported = false;
321 static TensorShape GetInputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCin in GetInputShape()
322 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape()
324 constexpr static const unsigned int inputSize = 48; // batchIn * heightIn * widthIn * channelIn
325 …onstexpr static const unsigned int outputSize = 48; // batchOut * heightOut * widthOut * channelOut
327 static IConnectableLayer* AddReceiverLayer(INetwork* network, in AddReceiverLayer()
340 static std::vector<IConnectableLayer*> AddConstantLayers(INetwork* network, in AddConstantLayers()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DSubtractionTestImpl.cpp107 LayerTestResult<float, 4> SubtractionTest( in SubtractionTest() function