Home
last modified time | relevance | path

Searched defs:inputDepth (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/
H A Dconvolution-operator-tester.h101 uint32_t inputDepth, in inputSize()
109 inline ConvolutionOperatorTester& inputDepth(uint32_t inputDepth) { in inputDepth() argument
115 inline uint32_t inputDepth() const { in inputDepth() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DAveragePool3d.cpp80 const int64_t inputDepth = input_.size(-3); in get_output_shape() local
110 const int64_t inputDepth = input.size(-3); in q_avg_pool3d() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DTransposeConvolution2d.cpp37 const unsigned int inputDepth = inputShape[channelsIndex]; in TransposeConvolution2dImpl() local
H A DConv3dImpl.cpp50 const unsigned int inputDepth = rInputShape[depthIndex]; in Convolve3d() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DPooling3dTestImpl.cpp62 unsigned int inputDepth = armnn::numeric_cast<unsigned int>(inputShape[depthIndex]); in SimplePooling3dTestImpl() local
170 unsigned int inputDepth = 3; in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon() local
1125 const unsigned int inputDepth = 48; in ComparePooling3dTestCommon() local
H A DConv3dTestImpl.cpp178 unsigned int inputDepth = armnn::numeric_cast<unsigned int>(inputShape[1]); in SimpleConvolution3dTestImpl() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DConv2D.cpp362 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in convQuant8PerChannelNhwc() local
455 [[maybe_unused]] uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in convQuant8PerChannelNhwc() local
DReshape.cpp127 const int inputDepth = extInputShape.Dims(3); in padGeneric() local
DDepthwiseConv2D.cpp288 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in depthwiseConvQuant8PerChannelNhwc() local
/aosp_15_r20/external/armnn/delegate/classic/src/
H A DPooling.hpp214 unsigned int inputDepth = inputTensorInfo.GetShape()[1]; in VisitPooling3dOperator() local
H A DConvolution.hpp327 const unsigned int inputDepth = inputTensorInfo.GetShape()[1]; in VisitConv3dOperator() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp2002 int64_t inputDepth,
2162 int64_t inputDepth,
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp1323 unsigned int inputDepth = inputTensorInfo.GetShape()[1]; in ParseConv3D() local
/aosp_15_r20/external/python/cpython2/Mac/Modules/qt/
D_Qtmodule.c13776 long inputDepth; in Qt_GraphicsExportGetInputImageDepth() local