aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/grappler/grappler_item_builder_test.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-05-31 21:24:02 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-05-31 21:27:22 -0700
commitd9620cab82760834a418df7d914c3c21b984b13d (patch)
treea2554d8eafa370d839790253e54e6748edd2c8e5 /tensorflow/core/grappler/grappler_item_builder_test.cc
parent25bb504ccd204f916441d2bebe8aa008e85e8433 (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.cc1
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);