Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc2910 absl::flat_hash_map<int, std::vector<Node*>> replicate_input_fan_in_nodes; in BuildExecuteNodes() local
2927 std::vector<Node*>& nodes = replicate_input_fan_in_nodes[e->dst_input()]; in BuildExecuteNodes()
3155 << replicate_input_fan_in_nodes[input_num][core]->name() in BuildExecuteNodes()
3157 graph->AddEdge(replicate_input_fan_in_nodes[input_num][core], 0, in BuildExecuteNodes()
3185 graph->AddEdge(replicate_input_fan_in_nodes[input_num][core], 0, in BuildExecuteNodes()