Searched refs:execute_nodes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 57 const absl::flat_hash_set<Node*>& execute_nodes, Node* node) { in IsExecuteNodeOrIdentityToExecuteNode() argument
58 if (execute_nodes.find(node) != execute_nodes.end()) return true; in IsExecuteNodeOrIdentityToExecuteNode()
67 if (!IsExecuteNodeOrIdentityToExecuteNode(graph, loop_nodes, execute_nodes, in IsExecuteNodeOrIdentityToExecuteNode()
94 const Node* enter_node, const absl::flat_hash_set<Node*> execute_nodes) { in ResourceOnlyUsedForTPUExecuteInLoop() argument
101 if (!IsExecuteNodeOrIdentityToExecuteNode(graph, loop_nodes, execute_nodes, in ResourceOnlyUsedForTPUExecuteInLoop()
|
| D | distributed_tpu_rewrite_pass.cc | 3057 std::vector<std::vector<Node*>> execute_nodes(params_info.NumReplicas()); in BuildExecuteNodes() local
3101 execute_nodes[replica].push_back(node); in BuildExecuteNodes()
3337 NodeOut{execute_nodes[replica][core_id], in BuildExecuteNodes()
3415 NodeOut{execute_nodes[0][core_id], in BuildExecuteNodes()
|