Searched defs:head_tail_outside_compilation_nodes (Results 1 – 1 of 1) sorted by relevance
336 std::map<string, std::vector<Node*>>* head_tail_outside_compilation_nodes) { in FindTaggedNodes()4681 const std::vector<Node*>& head_tail_outside_compilation_nodes) { in UpdateHeadTailOutsideCompilation()4715 const std::vector<Node*>& head_tail_outside_compilation_nodes, in RewriteTPUReplicateNode()4966 std::map<string, std::vector<Node*>> head_tail_outside_compilation_nodes; in InternalRun() local