Searched defs:conv_class (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
| H A D | test_quantize_pt2e_qat.py | 477 def __init__(self, conv_class, bn_class): argument
513 def __init__(self, conv_class, bn_class): argument
574 def __init__(self, conv_class): argument
608 def __init__(self, conv_class, bn_class): argument
674 def __init__(self, conv_class, bn_class): argument
695 def __init__(self, conv_class, bn_class, backbone): argument
952 conv_class = torch.nn.Conv1d variable in TestQuantizePT2EQAT_ConvBn1d
961 conv_class = torch.nn.Conv2d variable in TestQuantizePT2EQAT_ConvBn2d
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_efficient_conv_bn_eval.py | 34 conv_class, argument
58 conv_class, argument
|