/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...] |