Searched refs:from_example (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torchgen/ |
| D | gen_schema_utils.py | 32 def from_example(obj: Any) -> Union[BaseType, ListType, CustomClassType]: member in TypeGen
47 all_base_tys = [TypeGen.from_example(x) for x in obj]
63 def from_example( member in ReturnGen
66 return Return(name, TypeGen.from_example(obj), annotation)
71 def from_example( member in ArgumentGen
75 name, TypeGen.from_example(obj), default=default, annotation=annotation
81 def from_example( member in FunctionSchemaGen
88 args.append(ArgumentGen.from_example(name, inp, None, None))
94 ReturnGen.from_example(None, out, None) for out in example_outputs
|
| /external/pytorch/torch/_library/ |
| D | utils.py | 237 return FunctionSchemaGen.from_example(
|
| /external/pytorch/test/functorch/ |
| D | test_control_flow.py | 3988 ty = TypeGen.from_example(example_val)
3999 ty1 = TypeGen.from_example(li1)
4000 ty2 = TypeGen.from_example(li1)
4012 schema1 = FunctionSchemaGen.from_example("test_op1", inps, torch.ones(1))
4013 schema2 = FunctionSchemaGen.from_example(
4020 schema3 = FunctionSchemaGen.from_example(
|