Searched defs:inputDepth (Results 1 – 10 of 10) sorted by relevance
101 uint32_t inputDepth, in inputSize()109 inline ConvolutionOperatorTester& inputDepth(uint32_t inputDepth) { in inputDepth() argument115 inline uint32_t inputDepth() const { in inputDepth() function
80 const int64_t inputDepth = input_.size(-3); in get_output_shape() local110 const int64_t inputDepth = input.size(-3); in q_avg_pool3d() local
37 const unsigned int inputDepth = inputShape[channelsIndex]; in TransposeConvolution2dImpl() local
50 const unsigned int inputDepth = rInputShape[depthIndex]; in Convolve3d() local
214 unsigned int inputDepth = inputTensorInfo.GetShape()[1]; in VisitPooling3dOperator() local
327 const unsigned int inputDepth = inputTensorInfo.GetShape()[1]; in VisitConv3dOperator() local
62 unsigned int inputDepth = armnn::numeric_cast<unsigned int>(inputShape[depthIndex]); in SimplePooling3dTestImpl() local170 unsigned int inputDepth = 3; in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon() local1125 const unsigned int inputDepth = 48; in ComparePooling3dTestCommon() local
178 unsigned int inputDepth = armnn::numeric_cast<unsigned int>(inputShape[1]); in SimpleConvolution3dTestImpl() local
2002 int64_t inputDepth,2162 int64_t inputDepth,
1323 unsigned int inputDepth = inputTensorInfo.GetShape()[1]; in ParseConv3D() local