Searched refs:name_to_cost (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 164 const std::unordered_map<string, const CostGraphDef::Node*>& name_to_cost, in FindInputFeatures() argument
178 auto it = name_to_cost.find(input_node_name); in FindInputFeatures()
179 if (it == name_to_cost.end() || output_index < 0) { in FindInputFeatures()
290 std::unordered_map<string, const CostGraphDef::Node*> name_to_cost; in CostGraphToOpPerformanceData() local
293 name_to_cost[node.name()] = &node; in CostGraphToOpPerformanceData()
305 auto it = name_to_cost.find(node.name()); in CostGraphToOpPerformanceData()
306 if (it == name_to_cost.end()) { in CostGraphToOpPerformanceData()
315 FindInputFeatures(node, name_to_cost, name_to_node); in CostGraphToOpPerformanceData()
|
| D | utils.h | 43 const std::unordered_map<string, const CostGraphDef::Node*>& name_to_cost,
|
| D | graph_properties.cc | 2282 std::unordered_map<string, const CostGraphDef::Node*> name_to_cost; in InferFromCostGraph() local
2285 name_to_cost[node.name()] = &node; in InferFromCostGraph()
2302 auto it = name_to_cost.find(node.name()); in InferFromCostGraph()
2303 if (it == name_to_cost.end()) { in InferFromCostGraph()
2307 FindInputFeatures(node, name_to_cost, name_to_node); in InferFromCostGraph()
|