| 21 | namespace grappler { |
| 22 | |
| 23 | Status TFDataOptimizerBase::Optimize(Cluster* cluster, const GrapplerItem& item, |
| 24 | GraphDef* output) { |
| 25 | OptimizationStats stats; |
| 26 | Status s = OptimizeAndCollectStats(cluster, item, output, &stats); |
| 27 | if (s.ok() && stats.num_changes > 0) { |
| 28 | metrics::RecordTFDataOptimization(name(), stats.num_changes); |
| 29 | } |
| 30 | return s; |
| 31 | } |
| 32 | |
| 33 | } // namespace grappler |
| 34 | } // namespace tensorflow |