Searched defs:op_schema (Results 1 – 5 of 5) sorted by relevance
33 if (auto op_schema = node->maybeSchema()) { in run() local
730 const auto op_schema = fn.operator_schema(); in onFunctionExit() local
970 auto op_schema = node->getOperator().schema(); in process_ops_for_mobile() local
2930 def op_schema(cls, op) -> torch.FunctionSchema: member in ExtensionHandler
94 def op_schema(cls, op): member in TestSerialize.test_export_with_extension_op_serialization.FooExtensionHandler