diff options
author | 2018-03-28 11:18:45 -0700 | |
---|---|---|
committer | 2018-03-28 11:23:03 -0700 | |
commit | 95efef3271d67dd63ec2e397012a20d63d088668 (patch) | |
tree | 06da73bb411d37fee0d20be5707e4b3d733191af /tensorflow/core/grappler/optimizers/graph_optimizer_stage.h | |
parent | b1f6c4e0dee8732d8e25262052656e9cdf5ca513 (diff) |
Make ArithmeticOptimizer robust to failures of shape inference and individual stages.
Get rid of graph annotation and use GraphProperties directly.
PiperOrigin-RevId: 190801044
Diffstat (limited to 'tensorflow/core/grappler/optimizers/graph_optimizer_stage.h')
-rw-r--r-- | tensorflow/core/grappler/optimizers/graph_optimizer_stage.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/optimizers/graph_optimizer_stage.h b/tensorflow/core/grappler/optimizers/graph_optimizer_stage.h index be95c00d2d..c7af82abbb 100644 --- a/tensorflow/core/grappler/optimizers/graph_optimizer_stage.h +++ b/tensorflow/core/grappler/optimizers/graph_optimizer_stage.h @@ -117,6 +117,9 @@ class GraphOptimizerStage { : optimizer_name_(optimizer_name), stage_name_(stage_name), ctx_(ctx) {} virtual ~GraphOptimizerStage() = default; + const string& stage_name() const { return stage_name_; } + const string& optimizer_name() const { return optimizer_name_; } + // Check if we should try to simplify node. Returning true doesn't // guarantee that node will be simplified. // |