diff options
author | 2017-05-31 21:24:02 -0700 | |
---|---|---|
committer | 2017-05-31 21:27:22 -0700 | |
commit | d9620cab82760834a418df7d914c3c21b984b13d (patch) | |
tree | a2554d8eafa370d839790253e54e6748edd2c8e5 /tensorflow/core/grappler/grappler_item_builder_test.cc | |
parent | 25bb504ccd204f916441d2bebe8aa008e85e8433 (diff) |
Add flag to determine whether to do L1 optimizations and inline functions. Default is to do them. In tf_optimizer don't inline or do l1 optimizations.
PiperOrigin-RevId: 157673614
Diffstat (limited to 'tensorflow/core/grappler/grappler_item_builder_test.cc')
-rw-r--r-- | tensorflow/core/grappler/grappler_item_builder_test.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/grappler_item_builder_test.cc b/tensorflow/core/grappler/grappler_item_builder_test.cc index 54400f7051..92225ffb1b 100644 --- a/tensorflow/core/grappler/grappler_item_builder_test.cc +++ b/tensorflow/core/grappler/grappler_item_builder_test.cc @@ -70,6 +70,7 @@ std::unique_ptr<GrapplerItem> CreateGrapplerItem(const GraphDef &def, const CollectionDef &fetches) { MetaGraphDef meta_def; ItemConfig cfg; + cfg.inline_functions = true; *meta_def.mutable_graph_def() = def; (*meta_def.mutable_collection_def())["train_op"] = fetches; return GrapplerItemFromMetaGraphDef("0", meta_def, cfg); |