Searched refs:WhileLoopFrame (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_util.h | 49 struct WhileLoopFrame { struct
53 WhileLoopFrame* parent = nullptr; argument
79 std::unordered_map<string, WhileLoopFrame>* frames,
|
| D | functionalize_control_flow_util.cc | 54 std::unordered_map<string, WhileLoopFrame>* frames, in ExtractWhileLoopFrames()
66 WhileLoopFrame& frame = (*frames)[cf.frame_name]; in ExtractWhileLoopFrames()
67 WhileLoopFrame* parent = in ExtractWhileLoopFrames()
|
| D | functionalize_while.cc | 61 Status CopySubgraph(const Graph& graph, const WhileLoopFrame* frame, in CopySubgraph()
122 Status BuildLoopCondition(const Graph& graph, WhileLoopFrame* frame, in BuildLoopCondition()
158 Status BuildLoopBody(const Graph& graph, WhileLoopFrame* frame, in BuildLoopBody()
214 Status FunctionalizeLoop(Graph* graph, WhileLoopFrame* frame, in FunctionalizeLoop()
527 std::unordered_map<string, WhileLoopFrame> frames; in FunctionalizeWhileLoop()
532 std::deque<WhileLoopFrame*> worklist; in FunctionalizeWhileLoop()
543 WhileLoopFrame* frame = worklist.front(); in FunctionalizeWhileLoop()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 223 const std::unordered_map<string, WhileLoopFrame>& frames, in FindTPUCompileNodes()
226 std::deque<const WhileLoopFrame*> worklist; in FindTPUCompileNodes()
237 const WhileLoopFrame* frame = worklist.front(); in FindTPUCompileNodes()
433 std::unordered_map<string, WhileLoopFrame> frames; in DetectHostTrainingLoop()
|