Searched refs:node_costs (Results 1 – 4 of 4) sorted by relevance
40 int node_id, const Costs& node_costs, in AddCostNode() argument61 node->set_compute_cost(node_costs.execution_time.asMicroSeconds().count()); in AddCostNode()62 node->set_compute_time(node_costs.compute_time.asMicroSeconds().count()); in AddCostNode()63 node->set_memory_time(node_costs.memory_time.asMicroSeconds().count()); in AddCostNode()64 node->set_inaccurate(node_costs.inaccurate); in AddCostNode()163 Costs node_costs; in PredictCosts() local167 node_costs = node_estimator_->PredictCosts(op_context); in PredictCosts()169 if (node_costs.inaccurate) { in PredictCosts()171 if (node_costs.num_ops_with_unknown_shapes > 0) in PredictCosts()173 << node_costs.num_ops_with_unknown_shapes << " unknown shapes"; in PredictCosts()[all …]
746 bool VirtualScheduler::MarkCurrNodeExecuted(const Costs& node_costs) { in MarkCurrNodeExecuted() argument756 MultiplyCosts(node_costs, node_state.execution_count); in MarkCurrNodeExecuted()770 std::make_pair(node_costs.execution_time.asMicroSeconds().count(), in MarkCurrNodeExecuted()771 !node_costs.inaccurate); in MarkCurrNodeExecuted()
280 bool MarkCurrNodeExecuted(const Costs& node_costs);
1623 Costs node_costs = SimplePredictCosts(op_context); in RunScheduler() local1632 more_nodes = scheduler_->MarkCurrNodeExecuted(node_costs); in RunScheduler()