Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc299 Node* loop_switch_node; in GetLoopSwitchNode() local
314 Status GetOrCreateBeforeEachIterationNode(Graph* graph, Node* loop_switch_node, in GetOrCreateBeforeEachIterationNode()
347 Status AddNoOpAfterLastIteration(Graph* graph, Node* loop_switch_node, in AddNoOpAfterLastIteration()