experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[TF2XLA] Enable MLIR bridge usage in XlaCompiler::CompileGraph entry point

Browse Source 
PiperOrigin-RevId: 348513794
Change-Id: Ibbae0a765a959d6d61d60bf9ba6081b49df80f26
This commit is contained in:
Pankaj Kanwar 2020-12-21 12:54:19 -08:00 committed by TensorFlower Gardener
parent 428aef7972
commit 265454ac0f
1 changed files with 0 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tensorflow/compiler/tf2xla/xla_compiler.cc
View File

@ -1276,20 +1276,6 @@ Status XlaCompiler::CompileGraph(
CompilationResult* result) {
VLOG(1) << "Executing graph symbolically to populate XlaBuilder.: " << name;
absl::optional<ConfigProto> 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<XlaCompiler::Argument>(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(