Searched refs:next_iterations (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_while.cc | 170 std::vector<Node*> next_iterations; in BuildLoopBody() local
171 next_iterations.reserve(frame->args.size()); in BuildLoopBody()
201 next_iterations.push_back(arg.next_iteration); in BuildLoopBody()
208 TF_RETURN_IF_ERROR(CopySubgraph(graph, frame, std::move(next_iterations), in BuildLoopBody()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | deadness_analysis_test.cc | 962 std::array<Output, 2> next_iterations; in TEST() local
978 next_iterations[i] = ops::NextIteration( in TEST()
982 ->UpdateEdge(next_iterations[i].node(), 0, iv.output.node(), 1) in TEST()
999 EXPECT_NE(predicate_map[ControlOutputFor(next_iterations[0])], in TEST()
1000 predicate_map[ControlOutputFor(next_iterations[1])]); in TEST()
1001 EXPECT_NE(predicate_map[ControlOutputFor(next_iterations[0])], ""); in TEST()
1002 EXPECT_NE(predicate_map[ControlOutputFor(next_iterations[1])], ""); in TEST()
|