Home
last modified time | relevance | path

Searched refs:CustomClassType (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torchgen/
Dgen_schema_utils.py10 CustomClassType,
32 def from_example(obj: Any) -> Union[BaseType, ListType, CustomClassType]:
44 return CustomClassType(obj._type().name()) # type: ignore[attr-defined]
Dmodel.py1842 return CustomClassType(m.group(1))
1943 class CustomClassType(Type): class
/external/pytorch/tools/test/
Dtest_codegen_model.py15 CustomClassType,
155 CustomClassType, Type.parse(custom_class_name_with_prefix)
157 self.assertTrue(isinstance(custom_class_type, CustomClassType))