Home
last modified time | relevance | path

Searched defs:execute_nodes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dhost_training_loop_optimization_util.cc58 const absl::flat_hash_set<Node*>& execute_nodes, Node* node) { in IsExecuteNodeOrIdentityToExecuteNode()
95 const Node* enter_node, const absl::flat_hash_set<Node*> execute_nodes) { in ResourceOnlyUsedForTPUExecuteInLoop()
H A Ddistributed_tpu_rewrite_pass.cc3442 std::vector<std::vector<Node*>> execute_nodes(params_info.NumReplicas()); in BuildExecuteNodes() local