code: def _set_torch_reg_op(op_type, op_func):
      _TORCH_OPS_REG[op_type] = op_func
  
file_path: turicreate/src/external/coremltools_wrap/coremltools/coremltools/converters/mil/frontend/torch/test/test_custom_ops.py
imports: import itertools
  import numpy
  import pytest
  import torch
  import torch.nn
  import coremltools
  from .testing_utils import *
  from coremltools.converters.mil.mil.ops.defs._op_reqs import *
  from coremltools.converters.mil.frontend.torch.torch_op_registry import register_torch_op
  from coremltools.converters.mil.frontend.torch.torch_op_registry import _TORCH_OPS_REGISTRY
  from coremltools.converters.mil.frontend.torch.ops import _get_inputs
  from coremltools.converters.mil.mil import Builder
label_desc: Use image node_id 2 for calling a global function with example usage: _set_torch_reg_op(op_type, op_func) without return types