Searched refs:tensor_renaming (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.cc | 120 std::unordered_map<string, string> tensor_renaming; in GraphToFunctionDef() local
138 tensor_renaming[strings::StrCat(node->name(), ":0")] = normalized; in GraphToFunctionDef()
215 tensor_renaming[strings::StrCat(node->name(), ":", i)] = tensor_name; in GraphToFunctionDef()
245 const auto iter = tensor_renaming.find(node_def->input(i)); in GraphToFunctionDef()
246 if (iter == tensor_renaming.end()) { in GraphToFunctionDef()
264 const auto iter = tensor_renaming.find(return_value); in GraphToFunctionDef()
265 if (iter == tensor_renaming.end()) { in GraphToFunctionDef()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 163 const std::unordered_map<string, string>& tensor_renaming, in FillFunctionBody() argument
220 const auto iter = tensor_renaming.find(original_input_name); in FillFunctionBody()
221 if (iter == tensor_renaming.end()) { in FillFunctionBody()
327 std::unordered_map<string, string> tensor_renaming; in GraphToFunctionDef() local
355 tensor_renaming[strings::StrCat(node->name(), ":", idx)] = input_name; in GraphToFunctionDef()
385 if (tensor_renaming.find(original_name) == tensor_renaming.end()) { in GraphToFunctionDef()
386 tensor_renaming[original_name] = new_name; in GraphToFunctionDef()
393 FillFunctionBody(fn_name, node_names, body_nodes, tensor_renaming, fdef)); in GraphToFunctionDef()
402 const auto iter = tensor_renaming.find(return_value); in GraphToFunctionDef()
403 if (iter == tensor_renaming.end()) { in GraphToFunctionDef()
|