| 461 | } // namespace |
| 462 | |
| 463 | void CostModel::InitFromGraph(const Graph& g) { |
| 464 | const int num_node_ids = g.num_node_ids(); |
| 465 | slot_bytes_.reserve(num_node_ids); |
| 466 | count_.reserve(num_node_ids); |
| 467 | time_.reserve(num_node_ids); |
| 468 | max_mem_usage_.reserve(num_node_ids); |
| 469 | max_exec_time_.reserve(num_node_ids); |
| 470 | output_port_alloc_ids_.reserve(num_node_ids); |
| 471 | |
| 472 | AddNodesToCostModel(g, this); |
| 473 | AssignSizes(g, this); |
| 474 | EstimateComputationCosts(g, this); |
| 475 | CheckInitialized(g); |
| 476 | } |
| 477 | |
| 478 | void CostModel::AddToCostGraphDef(const Graph* graph, |
| 479 | CostGraphDef* cost_graph) const { |
no test coverage detected