Searched refs:wInput (Results 1 – 4 of 4) sorted by relevance
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | TransposeConvolution2dTestImpl.cpp | 232 constexpr unsigned int wInput = 3u; in SimpleTransposeConvolution2dTest() local
233 constexpr unsigned int hInput = wInput; in SimpleTransposeConvolution2dTest()
241 TensorShape inputShape = { batches, channels, hInput, wInput }; in SimpleTransposeConvolution2dTest()
321 constexpr unsigned int wInput = 4u; in PaddedTransposeConvolution2dTest() local
322 constexpr unsigned int hInput = wInput; in PaddedTransposeConvolution2dTest()
330 TensorShape inputShape = { batches, channels, hInput, wInput }; in PaddedTransposeConvolution2dTest()
412 constexpr unsigned int wInput = 3u; in StridedTransposeConvolution2dTest() local
413 constexpr unsigned int hInput = wInput; in StridedTransposeConvolution2dTest()
421 TensorShape inputShape = { batches, channels, hInput, wInput }; in StridedTransposeConvolution2dTest()
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | WorkloadDataValidation.cpp | 645 constexpr unsigned int wInput = 3u; variable
652 const TensorShape inputShape { nInput, cInput, hInput, wInput };
654 const TensorShape weightShape{ cOutput, cInput, hInput, wInput };
690 constexpr unsigned int wInput = 3u; variable
697 const TensorShape inputShape { nInput, cInput, hInput, wInput };
699 const TensorShape weightShape{ cOutput, cInput, hInput, wInput };
736 constexpr unsigned int wInput = 3u; variable
743 const TensorShape inputShape { nInput, cInput, hInput, wInput };
745 const TensorShape weightShape{ cOutput, cInput, hInput, wInput };
|
| D | TransposeConvolution2dEndToEndTestImpl.hpp | 52 constexpr unsigned int wInput = 3u; in TransposeConvolution2dEndToEnd() local
53 constexpr unsigned int hInput = wInput; in TransposeConvolution2dEndToEnd()
61 TensorShape inputShape = MakeTensorShape(batches, channels, hInput, wInput, dataLayout); in TransposeConvolution2dEndToEnd()
|
| /external/armnn/src/armnn/layers/ |
| D | TransposeConvolution2dLayer.cpp | 70 const unsigned int wInput = inputShape[dataLayoutIndex.GetWidthIndex()]; in InferOutputShapes() local
79 unsigned int wOutput = (wInput - 1) * m_Param.m_StrideX + wKernel - wPadding; in InferOutputShapes()
|