Searched refs:feed_remapping (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler_util.cc | 56 const std::unordered_map<string, string>& feed_remapping, in AddArgNodes() argument
64 const auto remap_it = feed_remapping.find(key); in AddArgNodes()
150 const std::unordered_map<string, string>& feed_remapping) { in RewriteAndPruneGraph() argument
156 TF_RETURN_IF_ERROR(AddArgNodes(graph, node_map, config.feed(), feed_remapping, in RewriteAndPruneGraph()
277 std::unordered_map<string, string> feed_remapping; in InitGraph() local
279 &feed_remapping, &first_copy_def)); in InitGraph()
287 for (const auto& entry : feed_remapping) { in InitGraph()
300 TF_RETURN_IF_ERROR(RewriteAndPruneGraph(g.get(), config, feed_remapping)); in InitGraph()
|
| D | tf2xla_util.h | 43 std::unordered_map<string, string>* feed_remapping, GraphDef* graph_def);
|
| D | tf2xla_util.cc | 334 std::unordered_map<string, string>* feed_remapping, GraphDef* graph_def) { in AddPlaceholdersForFeeds() argument
351 (*feed_remapping)[name_port] = info.placeholder_name; in AddPlaceholdersForFeeds()
|