Searched defs:execute_nodes (Results 1 – 2 of 2) sorted by relevance
57 const absl::flat_hash_set<Node*>& execute_nodes, Node* node) { in IsExecuteNodeOrIdentityToExecuteNode()94 const Node* enter_node, const absl::flat_hash_set<Node*> execute_nodes) { in ResourceOnlyUsedForTPUExecuteInLoop()
3057 std::vector<std::vector<Node*>> execute_nodes(params_info.NumReplicas()); in BuildExecuteNodes() local