code: def _try_to_transform(op1, op2, block):
      # replace the pow2(x) --> sqrt(x) with identity(x)
      x = mb.identity(x=op1.x, name=op2.outputs[0].name, before_op=op1)
  
      # update the graph
      op2.enclosing_block.replace_uses_of_var_after_op(
          anchor_op=op2, old_var=op2.outputs[0], new_var=x
      )
  
      # remove the ops
      block.remove_ops([op1, op2])
  
      return True
  
file_path: coremltools/coremltools/converters/mil/backend/mil/passes/fuse_pow2_sqrt.py
imports: from coremltools.converters.mil.mil import Builder
  from coremltools.converters.mil.mil.passes.graph_pass import AbstractGraphPass
  from coremltools.converters.mil.mil.passes.helper import block_context_manager
  from coremltools.converters.mil.mil.passes.pass_registry import register_pass
label_desc: Use image node_id 2 for calling a global function with example usage: _try_to_transform(op1, op2, block) and returns: True