Searched refs:name_to_node_map (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 68 std::unordered_map<string, NodeDef*>* name_to_node_map) { in GetNodeNameToNodeDefMap() argument
71 (*name_to_node_map)[node->name()] = node; in GetNodeNameToNodeDefMap()
88 const std::unordered_map<string, NodeDef*>& name_to_node_map, in GetReachableNodesAndVariables() argument
107 NodeDef* node = name_to_node_map.at(node_name); in GetReachableNodesAndVariables()
120 const std::unordered_map<string, NodeDef*>& name_to_node_map, in GetVariableNameToTensorMap() argument
132 NodeDef* node_def = name_to_node_map.at(node_name); in GetVariableNameToTensorMap()
185 std::unordered_map<string, NodeDef*> name_to_node_map; in FreezeGraphDef() local
186 GetNodeNameToNodeDefMap(&graph_def, &name_to_node_map); in FreezeGraphDef()
189 GetReachableNodesAndVariables(&graph_def, outputs, name_to_node_map, in FreezeGraphDef()
193 saved_model_bundle.session.get(), name_to_node_map, variable_node_names, in FreezeGraphDef()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 374 name_to_node_map = {
403 if is_engine_op and name_to_node_map[inp_name].op == "Const":
|