Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc501 std::unordered_map<string, Node*>* host_compute_key_placeholder_map) { in FindHostComputeKeyPlaceholderNodes() argument
502 host_compute_key_placeholder_map->clear(); in FindHostComputeKeyPlaceholderNodes()
504 (*host_compute_key_placeholder_map)[node->name()] = nullptr; in FindHostComputeKeyPlaceholderNodes()
513 auto iter = host_compute_key_placeholder_map->find(call_node_attr->s()); in FindHostComputeKeyPlaceholderNodes()
514 if (iter == host_compute_key_placeholder_map->end()) { in FindHostComputeKeyPlaceholderNodes()
4466 std::unordered_map<string, Node*> host_compute_key_placeholder_map; in Run() local
4468 graph, replicate_nodes, &host_compute_key_placeholder_map)); in Run()
4482 flr, host_compute_key_placeholder_map[node->name()], in Run()