| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| remove_nodes.cc | 66 std::map<string, string> inputs_to_rename; local
70 [&inputs_to_rename, &required_nodes, &any_nodes_removed](
93 inputs_to_rename[replace_node.name()] = target_name;
94 inputs_to_rename["^" + replace_node.name()] =
103 RenameNodeInputs(replaced_graph_def, inputs_to_rename,
|
| backports.cc | 67 std::map<string, string> inputs_to_rename; local
71 [&inputs_to_rename](const NodeMatch& match,
99 inputs_to_rename[tensor_array_v3_node.name() + ":1"] =
110 TF_RETURN_IF_ERROR(RenameNodeInputs(replaced_graph_def, inputs_to_rename,
|
| insert_logging.cc | 91 std::map<string, string> inputs_to_rename; local
138 inputs_to_rename[node.name() + ":0"] =
144 return RenameNodeInputs(logged_graph_def, inputs_to_rename,
|
| freeze_requantization_ranges.cc | 170 std::map<string, string> inputs_to_rename; local
187 inputs_to_rename[node.name() + ":0"] = min_node->name() + ":0";
197 inputs_to_rename[node.name() + ":1"] = max_node->name() + ":0";
203 return RenameNodeInputs(frozen_graph_def, inputs_to_rename,
|
| quantize_nodes.cc | 219 std::map<string, string> inputs_to_rename; local
233 inputs_to_rename[current_node->name() + ":*"] = original_name;
243 TF_RETURN_IF_ERROR(RenameNodeInputs(merged_graph_def, inputs_to_rename,
268 std::map<string, string> inputs_to_rename; local
279 [&inputs_to_rename, &graph_outputs](const NodeMatch& match,
285 inputs_to_rename[quantize_node.name() + ":0"] =
287 inputs_to_rename[quantize_node.name() + ":1"] =
289 inputs_to_rename[quantize_node.name() + ":2"] =
304 return RenameNodeInputs(replaced_graph_def, inputs_to_rename,
[all...] |
| transform_utils.h | 211 const std::map<string, string>& inputs_to_rename,
|
| transform_utils.cc | 476 const std::map<string, string>& inputs_to_rename,
481 for (const auto& input_to_rename : inputs_to_rename) {
|
