Searched defs:backbone (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/ |
| H A D | torchvision_models.py | 366 def __init__(self, backbone, classifier, aux_classifier=None): argument
453 arch_type, backbone, pretrained, progress, num_classes, aux_loss, **kwargs argument
|
| /aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
| H A D | test_quantize_pt2e_qat.py | 695 def __init__(self, conv_class, bn_class, backbone): argument
|