parser.emitError(parser.getNameLoc(), "unknown type: ") << keyword;
  return Type();
}

/// Print a type registered to this dialect.
void FallbackDialect::printType(Type type, DialectAsmPrinter &os) const {
  if (mlir::isa<TFTensorType>(type)) {
    os << "tf_tensor";
    return;
  }

  if (mlir::isa<TFAllocatorType>(type)) {
    os << "tf_allocator";
    return;
  }

  return mlir::Type();
}

// Print a type registered to this dialect.
void TensorflowMlrtDialect::printType(mlir::Type type,
                                      mlir::DialectAsmPrinter &os) const {
  if (mlir::isa<TFTensorType>(type)) {
    os << "tensor";
    return;
  }

  llvm_unreachable("unexpected tf_mlrt type kind");
}

}  // namespace tf_mlrt
}  // namespace tensorflow

  return mlir::Type();
}

// Print a type registered to this dialect.
void MlrtDialect::printType(mlir::Type type,
                            mlir::DialectAsmPrinter &os) const {
  if (mlir::isa<FutureType>(type)) {
    os << "future";
    return;
  }

  if (mlir::isa<PromiseType>(type)) {
    os << "promise";
    return;
  }