Home
last modified time | relevance | path

Searched refs:inputs_to_rename (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dremove_nodes.cc65 std::map<string, string> inputs_to_rename; in RemoveNodes() local
69 [&inputs_to_rename, &required_nodes, &any_nodes_removed]( in RemoveNodes()
92 inputs_to_rename[replace_node.name()] = target_name; in RemoveNodes()
93 inputs_to_rename["^" + replace_node.name()] = in RemoveNodes()
102 RenameNodeInputs(replaced_graph_def, inputs_to_rename, in RemoveNodes()
Dbackports.cc66 std::map<string, string> inputs_to_rename; in BackportTensorArrayV3Transform() local
70 [&inputs_to_rename](const NodeMatch& match, in BackportTensorArrayV3Transform()
98 inputs_to_rename[tensor_array_v3_node.name() + ":1"] = in BackportTensorArrayV3Transform()
109 TF_RETURN_IF_ERROR(RenameNodeInputs(replaced_graph_def, inputs_to_rename, in BackportTensorArrayV3Transform()
Dfreeze_requantization_ranges.cc169 std::map<string, string> inputs_to_rename; in FreezeRequantizationRanges() local
186 inputs_to_rename[node.name() + ":0"] = min_node->name() + ":0"; in FreezeRequantizationRanges()
196 inputs_to_rename[node.name() + ":1"] = max_node->name() + ":0"; in FreezeRequantizationRanges()
202 return RenameNodeInputs(frozen_graph_def, inputs_to_rename, in FreezeRequantizationRanges()
Dinsert_logging.cc90 std::map<string, string> inputs_to_rename; in InsertLogging() local
137 inputs_to_rename[node.name() + ":0"] = in InsertLogging()
143 return RenameNodeInputs(logged_graph_def, inputs_to_rename, in InsertLogging()
Dquantize_nodes.cc219 std::map<string, string> inputs_to_rename; in MergeDuplicateNodes() local
233 inputs_to_rename[current_node->name() + ":*"] = original_name; in MergeDuplicateNodes()
243 TF_RETURN_IF_ERROR(RenameNodeInputs(merged_graph_def, inputs_to_rename, in MergeDuplicateNodes()
268 std::map<string, string> inputs_to_rename; in RemoveRedundantQuantizations() local
279 [&inputs_to_rename, &graph_outputs](const NodeMatch& match, in RemoveRedundantQuantizations()
285 inputs_to_rename[quantize_node.name() + ":0"] = in RemoveRedundantQuantizations()
287 inputs_to_rename[quantize_node.name() + ":1"] = in RemoveRedundantQuantizations()
289 inputs_to_rename[quantize_node.name() + ":2"] = in RemoveRedundantQuantizations()
304 return RenameNodeInputs(replaced_graph_def, inputs_to_rename, in RemoveRedundantQuantizations()
Dtransform_utils.h211 const std::map<string, string>& inputs_to_rename,
Dtransform_utils.cc478 const std::map<string, string>& inputs_to_rename, in RenameNodeInputs() argument
483 for (const auto& input_to_rename : inputs_to_rename) { in RenameNodeInputs()