Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/qat/modules/
H A Dconv_fused.py457 _FLOAT_CONV_MODULE = nn.Conv1d variable in ConvBn1d
524 _FLOAT_CONV_MODULE = nn.Conv1d variable in ConvBnReLU1d
589 _FLOAT_CONV_MODULE = nn.Conv1d variable in ConvReLU1d
652 _FLOAT_CONV_MODULE = nn.Conv2d variable in ConvBn2d
721 _FLOAT_CONV_MODULE = nn.Conv2d variable in ConvBnReLU2d
786 _FLOAT_CONV_MODULE = nn.Conv2d variable in ConvReLU2d
849 _FLOAT_CONV_MODULE = nn.Conv3d variable in ConvBn3d
917 _FLOAT_CONV_MODULE = nn.Conv3d variable in ConvBnReLU3d
984 _FLOAT_CONV_MODULE = nn.Conv3d variable in ConvReLU3d
/aosp_15_r20/external/pytorch/torch/ao/nn/qat/modules/
H A Dconv.py140 _FLOAT_CONV_MODULE = nn.Conv1d variable in Conv1d
201 _FLOAT_CONV_MODULE = nn.Conv2d variable in Conv2d
265 _FLOAT_CONV_MODULE = nn.Conv3d variable in Conv3d