%ch2 = "tfrt_fallback_async.print_tensor"(%result_2, %ch1) : (!tfrt_fallback.tf_tensor, !tfrt.chain) -> !tfrt.chain

  tfrt.return %ch2 : !tfrt.chain
}

// CHECK-LABEL: --- Running 'batch_function_fallback_timeout_test'
func.func @batch_function_fallback_timeout_test() -> !tfrt.chain {
  %ch0 = tfrt.new.chain

  %a = tfrt_fallback_async.const_dense_tensor dense<[[4, 4], [4, 4]]> : tensor<2x2xi32>