Searched refs:symbolic_pythonop (Results 1 – 2 of 2) sorted by relevance
65 def symbolic_pythonop(g, *args, **kwargs): function78 register_custom_op_symbolic("prim::PythonOp", symbolic_pythonop, 1)
650 def symbolic_pythonop(g, *args, **kwargs): function653 torch.onnx.register_custom_op_symbolic("prim::PythonOp", symbolic_pythonop, 1)