self.assertEqual(quant_unit.node_name, 'Conv2D')
      self.assertRegex(quant_unit.func_name, r'^__inference_conv_\d+')
    elif target_opset == quant_opts_pb2.STABLEHLO:
      self.assertEqual(quant_unit.node_name, '_empty_node')
      self.assertRegex(
          quant_unit.func_name, r'^composite_conv_([a-zA-Z_0-9]+_)*fn_\d+'
      )
    else:
      assert False, f'Please add assertion for the op_set: {target_opset}.'