Home
last modified time | relevance | path

Searched defs:_ConvNd (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/_inductor/fx_passes/
Defficient_conv_bn_eval.py18 bn: nn.modules.batchnorm._BatchNorm, conv: nn.modules.conv._ConvNd, x: torch.Tensor
/external/pytorch/torch/ao/nn/qat/modules/
Dconv.py16 class _ConvNd(nn.modules.conv._ConvNd): class
/external/pytorch/torch/ao/nn/quantized/reference/modules/
Dconv.py22 class _ConvNd(torch.nn.modules.conv._ConvNd, ReferenceQuantizedModule): class
/external/pytorch/torch/nn/modules/
Dconv.py44 class _ConvNd(Module): class
/external/pytorch/torch/ao/nn/quantized/modules/
Dconv.py40 class _ConvNd(WeightedQuantizedModule): class