def register_custom_op_symbolic(symbolic_name, symbolic_fn, opset_version):
r"""
Registers ``symbolic_fn`` to handle ``symbolic_name``. See
"Custom Operators" in the module documentation for an example usage.
Args:
symbolic_name (str): The name of the custom operator in "<domain>::<op>"
format.
symbolic_fn (Callable): A function that takes in the ONNX graph and
the input arguments to the current operator, and returns new
operator nodes to add to the graph.
opset_version (int): The ONNX opset version in which to register.
"""
from torch.onnx import utils
utils.register_custom_op_symbolic(symbolic_name, symbolic_fn, opset_version)
def register_custom_op_symbolic(symbolic_name, symbolic_fn, opset_version):
from torch.onnx import utils
return utils.register_custom_op_symbolic(symbolic_name, symbolic_fn, opset_version)
