Searched defs:TorchModelType (Results 1 – 3 of 3) sorted by relevance
35 class TorchModelType(Enum): class201 model_type: Optional[TorchModelType] = None,229 def skip_dynamic_fx_test(reason: str, model_type: TorchModelType = None):
538 model_type: Optional[pytorch_test_common.TorchModelType] = None,576 model_type: Optional[pytorch_test_common.TorchModelType] = None,615 model_type: Optional[pytorch_test_common.TorchModelType] = None,
953 model_type: pytorch_test_common.TorchModelType,