Searched refs:op_cost (Results 1 – 3 of 3) sorted by relevance
696 std::map<string, Costs>* op_cost) { in FindOrCreateZero() argument697 auto it = op_cost->find(op_name); in FindOrCreateZero()698 if (it == op_cost->end()) { in FindOrCreateZero()701 it = op_cost->emplace(op_name, Costs::ZeroCosts()).first; in FindOrCreateZero()760 auto& op_cost = FindOrCreateZero(op_name, &op_to_cost_); in MarkCurrNodeExecuted() local761 op_cost = CombineCosts(op_cost, total_node_costs); in MarkCurrNodeExecuted()
470 int op_cost = 1; in PredictCwiseOp() local474 op_cost = it->second; in PredictCwiseOp()480 Costs costs = PredictOpCountBasedCost(op_count * op_cost, op_info); in PredictCwiseOp()1387 auto op_cost = PredictCosts(fused_op); in PredictFusedOp() local1389 fused_cost.compute_time += op_cost.compute_time; in PredictFusedOp()1390 fused_cost.inaccurate |= op_cost.inaccurate; in PredictFusedOp()1391 fused_cost.intermediate_memory_time += op_cost.intermediate_memory_time; in PredictFusedOp()
323 std::map<string, Costs>* op_cost);