Searched refs:head_tail_outside_compilation_nodes (Results 1 – 2 of 2) sorted by relevance
555 const std::vector<Node*>& head_tail_outside_compilation_nodes,579 const std::vector<Node*>& head_tail_outside_compilation_nodes);
320 std::map<string, std::vector<Node*>>* head_tail_outside_compilation_nodes) { in FindTaggedNodes() argument344 (*head_tail_outside_compilation_nodes)[cluster] = std::vector<Node*>(); in FindTaggedNodes()395 (*head_tail_outside_compilation_nodes)[cluster].push_back(node); in FindTaggedNodes()4180 const std::vector<Node*>& head_tail_outside_compilation_nodes) { in UpdateHeadTailOutsideCompilation() argument4181 for (Node* node : head_tail_outside_compilation_nodes) { in UpdateHeadTailOutsideCompilation()4214 const std::vector<Node*>& head_tail_outside_compilation_nodes, in RewriteTPUReplicateNode() argument4236 tf_device_assignment, head_tail_outside_compilation_nodes)); in RewriteTPUReplicateNode()4453 std::map<string, std::vector<Node*>> head_tail_outside_compilation_nodes; in Run() local4456 &head_tail_outside_compilation_nodes)); in Run()4484 head_tail_outside_compilation_nodes[node->name()], in Run()