Searched defs:node_costs (Results 1 – 5 of 5) sorted by relevance
41 int node_id, const Costs& node_costs, in AddCostNode()197 Costs node_costs; in PredictCosts() local
873 const NodeDef* node, const Costs& node_costs, const OpContext& op_context, in MarkNodeExecuted()1370 auto& node_costs = node_state.node_costs; in GetOrCalculateOutputSize() local1417 bool VirtualScheduler::MarkCurrNodeExecuted(const Costs& node_costs) { in MarkCurrNodeExecuted()
365 return [this, impl](const OpContext& op_context, NodeCosts* node_costs) { in OpLevelCostEstimator()653 NodeCosts node_costs; in PredictCosts() local1662 bool* found_unknown_shapes, NodeCosts* node_costs) { in PredictDefaultNodeCosts()
82 Costs node_costs; // Node costs per execution member
2293 Costs node_costs = SimplePredictCosts(op_context); in RunScheduler() local3186 Costs node_costs = Costs::ZeroCosts(false); in TEST_F() local