Searched refs:name_to_cost (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 165 const std::unordered_map<string, const CostGraphDef::Node*>& name_to_cost, in FindInputFeatures() argument
179 auto it = name_to_cost.find(input_node_name); in FindInputFeatures()
180 if (it == name_to_cost.end() || output_index < 0) { in FindInputFeatures()
291 std::unordered_map<string, const CostGraphDef::Node*> name_to_cost; in CostGraphToOpPerformanceData() local
294 name_to_cost[node.name()] = &node; in CostGraphToOpPerformanceData()
306 auto it = name_to_cost.find(node.name()); in CostGraphToOpPerformanceData()
307 if (it == name_to_cost.end()) { in CostGraphToOpPerformanceData()
316 FindInputFeatures(node, name_to_cost, name_to_node); in CostGraphToOpPerformanceData()
|
| D | utils.h | 44 const std::unordered_map<string, const CostGraphDef::Node*>& name_to_cost,
|
| D | graph_properties.cc | 2750 std::unordered_map<string, const CostGraphDef::Node*> name_to_cost; in InferFromCostGraph() local
2753 name_to_cost[node.name()] = &node; in InferFromCostGraph()
2770 auto it = name_to_cost.find(node.name()); in InferFromCostGraph()
2771 if (it == name_to_cost.end()) { in InferFromCostGraph()
2775 FindInputFeatures(node, name_to_cost, name_to_node); in InferFromCostGraph()
|