Searched refs:node_costs (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.h | 118 NodeCosts* node_costs) const;
122 NodeCosts* node_costs) const;
135 NodeCosts* node_costs) const;
137 NodeCosts* node_costs) const;
139 NodeCosts* node_costs) const;
141 NodeCosts* node_costs) const;
143 NodeCosts* node_costs) const;
145 NodeCosts* node_costs) const;
147 NodeCosts* node_costs) const;
149 NodeCosts* node_costs) const;
[all …]
|
| D | op_level_cost_estimator.cc | 364 return [this, impl](const OpContext& op_context, NodeCosts* node_costs) { in OpLevelCostEstimator() argument
365 return (this->*impl)(op_context, node_costs); in OpLevelCostEstimator()
650 NodeCosts node_costs; in PredictCosts() local
651 if (PredictNodeCosts(op_context, &node_costs).ok()) { in PredictCosts()
652 if (node_costs.has_costs) { in PredictCosts()
653 return node_costs.costs; in PredictCosts()
656 if (node_costs.minimum_cost_op) { in PredictCosts()
671 node_costs.num_compute_ops, node_costs.num_total_read_bytes(), in PredictCosts()
672 node_costs.num_total_write_bytes(), op_context.op_info); in PredictCosts()
677 costs.max_memory = node_costs.max_memory; in PredictCosts()
[all …]
|
| D | analytical_cost_estimator.cc | 40 int node_id, const Costs& node_costs, in AddCostNode() argument
61 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_temporary_memory_size(node_costs.temporary_memory); in AddCostNode()
65 node->set_persistent_memory_size(node_costs.persistent_memory); in AddCostNode()
66 node->set_inaccurate(node_costs.inaccurate); in AddCostNode()
191 Costs node_costs; in PredictCosts() local
195 node_costs = node_estimator_->PredictCosts(op_context); in PredictCosts()
197 if (node_costs.inaccurate) { in PredictCosts()
[all …]
|
| D | virtual_scheduler.h | 80 Costs node_costs; // Node costs per execution member
82 return MultiplyCosts(node_costs, execution_count); in TotalNodeCosts()
374 const Costs& node_costs,
484 virtual bool MarkCurrNodeExecuted(const Costs& node_costs);
|
| D | virtual_scheduler.cc | 868 const NodeDef* node, const Costs& node_costs, const OpContext& op_context) { in MarkNodeExecuted() argument
882 node_state.node_costs = node_costs; in MarkNodeExecuted()
898 !node_costs.inaccurate); in MarkNodeExecuted()
1352 bool VirtualScheduler::MarkCurrNodeExecuted(const Costs& node_costs) { in MarkCurrNodeExecuted() argument
1356 node, node_costs, in MarkCurrNodeExecuted()
|
| D | virtual_scheduler_test.cc | 2286 Costs node_costs = SimplePredictCosts(op_context); in RunScheduler() local
2295 more_nodes = scheduler_->MarkCurrNodeExecuted(node_costs); in RunScheduler()
|