Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDepthwiseConv2d.cu290 const int depthwiseMultiplier, in conv_depthwise2d_grad_weight_kernel()
404 int64_t depthwiseMultiplier = outputChannels / inputChannels; in conv_depthwise2d_forward_out() local
486 int depthwiseMultiplier = outputChannels / inputChannels; in conv_depthwise2d_backward_out() local
612 int depthwiseMultiplier = outputChannels / inputChannels; in conv_depthwise2d_grad_weight_out() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/operation_converters/
DDepthwiseConv2DOperationConverter.cpp80 int32_t depthwiseMultiplier = context->getConstantScalar<int32_t>(depthwiseMultiplierOperand); in convert() local