//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<[]>,

03\09\07;7\03\03\03\01\05\04=\03\05\03\11\1D\1B\05\03\03\07\03\03\03\00\05\04\1F\03\01\06\03\01\05\01\00\9E\07G\1B)\11\0B!\1B\1D\05\1B\1B\03\0F%\1F/!!)#\1F\19)\1F\13\15\1D\15G\11\1F\15\11\0F\0B\11builtin\00vhlo\00module\00func_v1\00return_v1\00custom_call_v1\00call_v1\00xla_call_module_serialization.mlir\00arg_attrs\00function_type\00res_attrs\00sym_name\00sym_visibility\00mhlo.num_partitions\00api_version\00backend_config\00call_target_name\00called_computations\00has_side_effect\00operand_layo...

9F\17\09\03\15\05\07\09\0BE\1D\19\0B\03\0B\0DG#\1F\0D\03\0D\05\03\05\0F\04%\03\0D\06\03\01\05\01\00^\04+\0F\0F!/\0B\0D\033\1D9\19\15\1F\1F\17-\19\11\0F\0B\11builtin\00vhlo\00module\00func_v1\00constant_v1\00get_dimension_size_v1\00compare_v1\00custom_call_v1\00dot_general_v1\00return_v1\00/tmp/t.mlir\00jax.uses_shape_polymorphism\00error_message\00mhlo.frontend_attributes\00\00false\00main\00Shape assertion failed\00shape_assertion\00grad_x\00grad_y\00\08A\0F\05\01\01\0B'1'3)\035\037\059;\11?)A'C'''\0B-KM-O",...