%0 = mhlo.custom_call @custom_call.my_custom_op(%arg0) {
    api_version = 4 : i32,
    backend_config = {foo = "bar"}
  } : (tensor<1x4xf32>) -> (tensor<1x8xf32>)
  func.return %0 : tensor<1x8xf32>

  //       CHECK: %0 = mhlo.custom_call @custom_call.my_custom_op(%arg0) {
  //  CHECK-SAME:   api_version = 4 : i32,
  //  CHECK-SAME:   backend_config = {foo = "bar"}

//CHECK: return %arg0 : tensor<4xi8>
//CHECK:}

func.func @custom_tf_op(%arg0: tensor<1x320x1x1xf32>, %arg1: tensor<2xi32>) -> tensor<1x1600x1x1xf32> {
  %0 = "vhlo.custom_call_v1" (%arg0, %arg1) <{api_version = #vhlo<api_version_v1 API_VERSION_ORIGINAL>, 
    backend_config = #vhlo.string_v1<"">,
    call_target_name = #vhlo.string_v1<"tf.ResizeBilinear">,
    called_computations = #vhlo.array_v1<[]>,

    //  func.func @main(%arg0: tensor<?xi32> {jax.arg_info = "x", mhlo.sharding = "{replicated}"}, %arg1: tensor<*xi32>) -> (tensor<?xi32> {jax.result_info = ""}) {
    //    stablehlo.custom_call @tf.call_tf_function(%arg0, %arg1) {api_version = 2 : i32, has_side_effect = true, tf.backend_config = {called_func = @_tf_func}} : (tensor<?xi32>, tensor<*xi32>) -> ()
    //    return %arg0 : tensor<?xi32>
    //  }

    // CHECK-SAME: {
    // CHECK-SAME:  api_version = 2 : i32,
    // CHECK-SAME:  has_side_effect = true,
    // CHECK-SAME:  tf.backend_config = {called_func = @_tf_func}
    // CHECK-SAME: }
    stablehlo.custom_call @tf.call_tf_function(%arg0, %arg1) {api_version = 2 : i32, has_side_effect = true, tf.backend_config = {called_func = @_tf_func}} : (tensor<?xi32>, tensor<*xi32>) -> ()

    stablehlo.custom_call @tf.call_tf_function(%arg0, %arg1) {api_version = 2 : i32, has_side_effect = true, tf.backend_config = {called_func = @_tf_func}} : (tensor<?xi32>, tensor<*xi32>) -> ()
    %arg2 = func.call @_stablehlo_f(%arg0) : (tensor<?xi32>) -> (tensor<?xi32>)
    return %arg2 : tensor<?xi32>

    cc_api_version = 2,
    make_default_target_header_only = True,
)

# copybara:uncomment_begin(google-only)
# py_proto_library(
#     name = "calibration_statistics_py_pb2",
#     api_version = 2,
#     deps = [
#         ":calibration_statistics_proto",
#     ],
# )
# copybara:uncomment_end

tf_cc_test(
    name = "calibration_statistics_collector_test",
    size = "small",

{
  "kind": "APIVersions",
  "apiVersion": "v1",
  "versions": [
    "versionsValue"
  ],
  "serverAddressByClientCIDRs": [
    {
      "clientCIDR": "clientCIDRValue",
      "serverAddress": "serverAddressValue"
    }
  ]

apiVersion: v1
kind: APIVersions
serverAddressByClientCIDRs:
- clientCIDR: clientCIDRValue
  serverAddress: serverAddressValue
versions:

{
  "kind": "APIVersions",
  "apiVersion": "v1",
  "versions": [
    "versionsValue"
  ],
  "serverAddressByClientCIDRs": [
    {
      "clientCIDR": "clientCIDRValue",
      "serverAddress": "serverAddressValue"
    }
  ]

{
  "kind": "APIVersions",
  "apiVersion": "v1",
  "versions": [
    "versionsValue"
  ],
  "serverAddressByClientCIDRs": [
    {
      "clientCIDR": "clientCIDRValue",
      "serverAddress": "serverAddressValue"
    }
  ]