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

Enable graph-as-function for exporting in bridge pass.

Browse Source 
PiperOrigin-RevId: 278626392
Change-Id: Iedacd0abcb91b052df076a416448d06fb9d27227
This commit is contained in:
Andy Ly 2019-11-05 08:02:56 -08:00 committed by TensorFlower Gardener
parent 51296ec43c
commit 87c5874443
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/tf2xla/mlir_bridge_pass.cc
View File

@ -96,6 +96,7 @@ Status MlirBridgePass::Run(const GraphOptimizationPassOptions& options) {
specs.graph_as_function = true;
GraphExportConfig confs;
confs.graph_as_function = true;
TF_ASSIGN_OR_RETURN(auto module,
ConvertGraphToMlir(**options.graph, debug_info,
*options.flib_def, specs, &context));