Add a flag to erase "_noinline" attribute to allow total inlining in Grappler.

PiperOrigin-RevId: 171722354

1 files changed, 9 insertions, 11 deletions

diff --git a/tensorflow/core/grappler/grappler_item_builder.h b/tensorflow/core/grappler/grappler_item_builder.h
index 4ce5055e7a..9a7f52228b 100644
--- a/tensorflow/core/grappler/grappler_item_builder.h
+++ b/tensorflow/core/grappler/grappler_item_builder.h
@@ -27,24 +27,22 @@ class MetaGraphDef;
 namespace grappler {
 
 struct ItemConfig {
-  ItemConfig()
-      : ignore_user_placement(true),
-        ignore_colocation(true),
-        placeholder_unknown_output_shape_dim(-1),
-        apply_optimizations(false),
-        inline_functions(false) {}
+  ItemConfig() {}
 
   // If true, ignore all user specified node placement.
-  bool ignore_user_placement;
+  bool ignore_user_placement = true;
   // If true, ignore all user specified colocation attributes.
-  bool ignore_colocation;
+  bool ignore_colocation = true;
   // Dimension to use if a placeholder node has an _output_shapes attribute with
   // a dimension of -1.
-  int placeholder_unknown_output_shape_dim;
+  int placeholder_unknown_output_shape_dim = -1;
   // If true, does L1 optimizations.
-  bool apply_optimizations;
+  bool apply_optimizations = false;
   // If true, does inlining.
-  bool inline_functions;
+  bool inline_functions = false;
+  // If true, erases all "_noinline" attributes from user-defined functions.
+  // Has no effect if "inline_functions" is disabled.
+  bool erase_noinline_attributes = false;
   // If non-empty, override the directory of asset paths.
   string assets_directory_override;
 };