| /external/tensorflow/tensorflow/compiler/xla/service/ |
| while_loop_analysis_test.cc | 55 HloInstruction* while_op = module->entry_computation()->root_instruction(); local
56 EXPECT_EQ(*ComputeWhileLoopTripCountUpperBound(while_op), 1);
86 HloInstruction* while_op = module->entry_computation()->root_instruction(); local
87 EXPECT_EQ(ComputeWhileLoopTripCountUpperBound(while_op), absl::nullopt);
119 HloInstruction* while_op = module->entry_computation()->root_instruction(); local
120 EXPECT_EQ(*ComputeWhileLoopTripCountUpperBound(while_op), 42);
|
| while_loop_simplifier_test.cc | 168 auto* while_op = computation->root_instruction(); local
169 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile);
170 auto* true_op = while_op->while_body()->AddInstruction(
173 while_op->while_body()->root_instruction()));
185 auto* while_op = computation->root_instruction(); local
186 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile);
187 auto* while_body = while_op->while_body();
201 auto* while_op = computation->root_instruction(); local
202 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile);
203 auto* while_body = while_op->while_body()
219 auto* while_op = computation->root_instruction(); local
[all...] |
| heap_simulator_test.cc | 80 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( local
93 schedule.set_sequence(entry_computation, {iter, data, tuple, while_op});
721 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( local
726 {param, while_op, body_param, cond_param, cond_iter, cond_data, cond_lt});
728 // The entry computation param and while_op are allocated first.
732 {kAlloc, tracker.BufferAt(while_op, {})},
733 {kAlloc, tracker.BufferAt(while_op, {0})},
734 {kAlloc, tracker.BufferAt(while_op, {1})},
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| tf2xla_util_test.cc | 357 auto while_op = local
391 auto while_op = local
|
| functionalize_control_flow_test.cc | 268 auto while_op = local
271 auto sink = ops::Identity(scope.WithOpName("sink"), while_op[0]);
420 auto while_op = local
540 auto while_op = local
689 auto while_op = local
692 auto sink_x = ops::Identity(scope.WithOpName("sink_x"), while_op[0]);
693 auto sink_y = ops::Identity(scope.WithOpName("sink_y"), while_op[1]);
939 auto while_op = ops::While(scope.WithOpName("outer\/LoopCond"), local
995 auto while_op = local
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| extract_outside_compilation_pass_test.cc | 736 auto while_op = local
739 ops::_Retval retval(s.WithOpName("retval"), while_op.output[0], 0);
[all...] |
