// it is the responsibility of the caller, typically required when the registry
// is appended to the context in a parallel context, which does not allow for
// extensions to be added.
inline void RegisterAllTensorFlowDialectsImpl(DialectRegistry &registry,
                                              bool include_extensions = true) {
  registry
      .insert<mlir::arith::ArithDialect, mlir::func::FuncDialect,