def ExtractTPUCopyWithDynamicShapeOpPass : Pass<"tf-extract-tpu-copy-with-dynamic-shape-op", "mlir::func::FuncOp"> {
  let summary = "Extract the TPUCopyWithDynamicShapeOp out of the host launch and place it on device launch";

  let description = [{
    This pass looks for TPUCopyWithDynamicShapeOp which wraps in a
    `tf_device.launch` with host device attribute. It extracts the ops and wrap