From 265454ac0f1666884545d2ce9f5e9c13cca3f67a Mon Sep 17 00:00:00 2001 From: Pankaj Kanwar Date: Mon, 21 Dec 2020 12:54:19 -0800 Subject: [PATCH] [TF2XLA] Enable MLIR bridge usage in XlaCompiler::CompileGraph entry point PiperOrigin-RevId: 348513794 Change-Id: Ibbae0a765a959d6d61d60bf9ba6081b49df80f26 --- tensorflow/compiler/tf2xla/xla_compiler.cc | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/tensorflow/compiler/tf2xla/xla_compiler.cc b/tensorflow/compiler/tf2xla/xla_compiler.cc index b03ac8f89fb..8dc8da7efed 100644 --- a/tensorflow/compiler/tf2xla/xla_compiler.cc +++ b/tensorflow/compiler/tf2xla/xla_compiler.cc @@ -1276,20 +1276,6 @@ Status XlaCompiler::CompileGraph( CompilationResult* result) { VLOG(1) << "Executing graph symbolically to populate XlaBuilder.: " << name; - absl::optional config_proto; - MlirBridgeRolloutPolicy policy = - GetMlirBridgeRolloutPolicy(*graph, config_proto); - if (policy == MlirBridgeRolloutPolicy::kEnabledByUser) { - VLOG(1) << "Using MLIR bridge"; - GraphDebugInfo debug_info; - TF_RETURN_IF_ERROR(CompileGraphToXlaHlo( - std::move(*graph), mlir::SpanToArrayRef(args), - {}, options_.device_type.type_string(), options.use_tuple_arg, - *options_.flib_def, debug_info, options_.shape_representation_fn, - result)); - return Status::OK(); - } - TF_RETURN_IF_ERROR(PropagateConstIntoFunctionalNodes( graph.get(), options_.flib_def, local_flib_def_.get())); TF_RETURN_IF_ERROR(RearrangeFunctionArguments(