Searched refs:feed_remapping (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla.cc | 68 const std::unordered_map<string, string>& feed_remapping) { in AddArgNodes() argument
75 const auto remap_it = feed_remapping.find(key); in AddArgNodes()
157 const std::unordered_map<string, string>& feed_remapping) { in RewriteAndPruneGraph() argument
163 AddArgNodes(graph, node_map, config.feed(), feed_remapping)); in RewriteAndPruneGraph()
352 std::unordered_map<string, string> feed_remapping; in InitGraph() local
354 &feed_remapping, &first_copy_def)); in InitGraph()
367 TF_RETURN_IF_ERROR(RewriteAndPruneGraph(g.get(), config, feed_remapping)); in InitGraph()
|
| D | tf2xla_util.h | 42 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()
|