Lines Matching defs:LazyIrSchema

102 def node_ctor_inputs(schema: LazyIrSchema) -> str:
113     schema: LazyIrSchema,
136 def aten_symbol(schema: LazyIrSchema) -> str:
188     def lowering_function(self, schema: LazyIrSchema) -> str:
191     def create_function(self, schema: LazyIrSchema, node_ctor_args: str) -> str:
194     def can_be_reused_function(self, schema: LazyIrSchema, node_ctor_args: str) -> str:
199     def node_base_ctor_call(self, schema: LazyIrSchema) -> str:
241     def gen(self, schema: LazyIrSchema) -> list[str]:
342     def lowering_function(self, schema: LazyIrSchema) -> str:
358     def create_function(self, schema: LazyIrSchema, node_ctor_args: str) -> str:
368     def can_be_reused_function(self, schema: LazyIrSchema, node_ctor_args: str) -> str:
415     def lazy_tensor_decls(self, func: NativeFunction, schema: LazyIrSchema) -> str:
463         schema: LazyIrSchema,
473     def metrics(self, func: NativeFunction, schema: LazyIrSchema) -> str:
476     def get_device(self, func: NativeFunction, schema: LazyIrSchema) -> str:
494     def shape_inference(self, func: NativeFunction, schema: LazyIrSchema) -> str:
569     def build_ir_node(self, func: NativeFunction, schema: LazyIrSchema) -> str:
589     def return_aten_tensor(self, func: NativeFunction, schema: LazyIrSchema) -> str: