Searched refs:loop_nodes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 56 const Graph& graph, const std::unordered_set<Node*>& loop_nodes, // NOLINT in IsExecuteNodeOrIdentityToExecuteNode() argument
59 if (loop_nodes.find(node) == loop_nodes.end()) return false; in IsExecuteNodeOrIdentityToExecuteNode()
67 if (!IsExecuteNodeOrIdentityToExecuteNode(graph, loop_nodes, execute_nodes, in IsExecuteNodeOrIdentityToExecuteNode()
93 const Graph& graph, const std::unordered_set<Node*>& loop_nodes, // NOLINT in ResourceOnlyUsedForTPUExecuteInLoop() argument
101 if (!IsExecuteNodeOrIdentityToExecuteNode(graph, loop_nodes, execute_nodes, in ResourceOnlyUsedForTPUExecuteInLoop()
114 const std::unordered_set<Node*>& loop_nodes, // NOLINT in ExtractExecuteNodeInfo() argument
184 graph, loop_nodes, it.first, it.second)); in ExtractExecuteNodeInfo()
255 host_training_loop_info.loop_nodes = frame->nodes; in FindTPUCompileNodes()
453 ExtractExecuteNodeInfo(compile_node, *graph, host_loop_info.loop_nodes, in AddReshardOp()
|
| D | host_training_loop_optimization_util.h | 57 std::unordered_set<Node*> loop_nodes; // NOLINT member
|