Searched defs:bn_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
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
954 bn_class = torch.nn.BatchNorm1d variable in TestQuantizePT2EQAT_ConvBn1d
963 bn_class = torch.nn.BatchNorm2d variable in TestQuantizePT2EQAT_ConvBn2d
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_efficient_conv_bn_eval.py | 35 bn_class, argument
59 bn_class, argument
|