def ConvertLaunchFuncToTFCallPass : Pass<"tf-device-convert-launch-func-to-tf-call", "ModuleOp"> {
  let summary = "Rewrites tf_device::LaunchFuncOp to TF::PartitionedCallOp";

  let description = [{
    This pass converts tf_device::LaunchFuncOp into an equivalent
    TF::PartitionedCallOp so that it can be exported to TensorFlow GraphDef.
  }];

  let constructor = "TFDevice::CreateConvertLaunchFuncToTFCallPass()";
}

func.func @unhandled_partitioned_call(%arg0: tensor<*xi32>, %arg1: tensor<*xi32>) -> (tensor<i32>, tensor<i32>) {
  // The argument types don't match the parameter types for the
  // pcall_multi_in_out function. That's fine for a PartitionedCallOp but not
  // for a standard CallOp, so this op can't be lowered.
  // CHECK: "tf.PartitionedCall"