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

[TF:MLIR] When importing GraphDef to MLIR for tfcompile, set upgrade_legacy flag

Browse Source 
to allow the functionalization of TensorFlow V1 control flow.

PiperOrigin-RevId: 295242525
Change-Id: Icf22382c788cd6da2a65cfd8b6a349e11ce8c489
This commit is contained in:
Bixia Zheng 2020-02-14 15:39:41 -08:00 committed by TensorFlower Gardener
parent b380277f3a
commit 1e74a754af
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/tf2xla/mlir_tf2xla.cc
View File

@ -110,7 +110,7 @@ Status ConvertGraphDefToXlaViaMlir(GraphDef graph_def,
specs.prune_unused_nodes = false;
specs.convert_legacy_fed_inputs = false;
specs.graph_as_function = false;
specs.upgrade_legacy = false;
specs.upgrade_legacy = true;
TF_RETURN_IF_ERROR(ConvertInputInfo(config, feed_name_remap, &specs));
TF_RETURN_IF_ERROR(ConvertOutputInfo(config, &specs));