Home
last modified time | relevance | path

Searched defs:kernelChannels (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_backward_spatial_convolutions.h125 const TensorIndex kernelChannels = variable
430 const TensorIndex kernelChannels = variable
H A Deigen_backward_cuboid_convolutions.h119 const TensorIndex kernelChannels = variable
442 const TensorIndex kernelChannels = variable
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DConv2dTestImpl.cpp230 unsigned int kernelChannels = armnn::numeric_cast<unsigned int>(originalKernelShape[1]); in SimpleConvolution2dTestImpl() local
402 unsigned int kernelChannels = armnn::numeric_cast<unsigned int>(kernelShape[3]); in SimpleConvolution2dNhwcTestImpl() local
1717 unsigned int kernelChannels = armnn::numeric_cast<unsigned int>(kernelShape[3]); in DepthwiseConvolution2dAsymmetricTestImpl() local
2274 unsigned int kernelChannels = armnn::numeric_cast<unsigned int>(originalKernelShape[3]); in DepthwiseConvolution2dTestImpl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDepthwiseConv2d.cu289 const int kernelChannels, in conv_depthwise2d_grad_weight_kernel()