Searched refs:op_cost (Results 1 – 2 of 2) sorted by relevance
57 std::map<string, Costs>* op_cost) { in FindOrCreateZero() argument58 auto it = op_cost->find(op_name); in FindOrCreateZero()59 if (it == op_cost->end()) { in FindOrCreateZero()62 it = op_cost->emplace(op_name, Costs::ZeroCosts()).first; in FindOrCreateZero()889 auto& op_cost = FindOrCreateZero(op_name, &op_to_cost_); in MarkNodeExecuted() local890 op_cost = CombineCosts(op_cost, total_node_costs); in MarkNodeExecuted()
792 int op_cost = 1; in PredictCwiseOp() local795 op_cost = it->second; in PredictCwiseOp()800 return PredictDefaultNodeCosts(op_count * op_cost, op_context, in PredictCwiseOp()