| 2224 | } // namespace |
| 2225 | |
| 2226 | Status LayoutOptimizer::Tune(const GrapplerItem& item, |
| 2227 | const GraphProperties& graph_properties, |
| 2228 | const TuningConfig& config, GraphDef* output) { |
| 2229 | auto status = graph_properties.AnnotateOutputShapes(output); |
| 2230 | if (!status.ok()) { |
| 2231 | VLOG(1) << "Annotate shape return status: " << status.ToString(); |
| 2232 | *output = item.graph; |
| 2233 | return status; |
| 2234 | } |
| 2235 | NodeMap node_map(output); |
| 2236 | DataLayoutOptimizer layout_optimizer(graph_properties, *virtual_placer_, |
| 2237 | config, nodes_to_preserve_, output, |
| 2238 | &node_map); |
| 2239 | status = layout_optimizer.Optimize(); |
| 2240 | return status; |
| 2241 | } |
| 2242 | |
| 2243 | Status LayoutOptimizer::Optimize(Cluster* cluster, const GrapplerItem& item, |
| 2244 | GraphDef* output) { |
nothing calls this directly
no test coverage detected