aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/grappler/optimizers/graph_optimizer_stage.h
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-04-20 16:05:47 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-20 16:08:25 -0700
commita0071844d0af47f22ab512363b56383acf762dff (patch)
tree1a35ee263240d9dbffa96c22f854d8745651b093 /tensorflow/core/grappler/optimizers/graph_optimizer_stage.h
parentc015a45646029f8c116028505f2da9e023b5c2b7 (diff)
Remove protected data members from GraphOptimizerStage.
PiperOrigin-RevId: 193737654
Diffstat (limited to 'tensorflow/core/grappler/optimizers/graph_optimizer_stage.h')
-rw-r--r--tensorflow/core/grappler/optimizers/graph_optimizer_stage.h5
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/optimizers/graph_optimizer_stage.h b/tensorflow/core/grappler/optimizers/graph_optimizer_stage.h
index ed398525f3..089cad36e9 100644
--- a/tensorflow/core/grappler/optimizers/graph_optimizer_stage.h
+++ b/tensorflow/core/grappler/optimizers/graph_optimizer_stage.h
@@ -182,7 +182,10 @@ class GraphOptimizerStage {
return ::tensorflow::grappler::AddEmptyNode(ctx_, name);
}
- protected: // Data members
+ protected:
+ const GraphOptimizerContext& ctx() const { return ctx_; }
+
+ private: // Data members
const string optimizer_name_;
const string stage_name_;
const GraphOptimizerContext ctx_;