Searched refs:FunctionSchemaGen (Results 1 – 3 of 3) sorted by relevance
205 from torchgen.gen_schema_utils import FunctionSchemaGen237 return FunctionSchemaGen.from_example(
79 class FunctionSchemaGen: class
4005 from torchgen.gen_schema_utils import FunctionSchemaGen4012 schema1 = FunctionSchemaGen.from_example("test_op1", inps, torch.ones(1))4013 schema2 = FunctionSchemaGen.from_example(4020 schema3 = FunctionSchemaGen.from_example(