/external/tensorflow/tensorflow/compiler/xla/service/ |
D | while_loop_invariant_code_motion_test.cc | 86 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 145 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 199 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 239 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 282 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 335 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 384 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 433 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 461 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
|
D | copy_insertion_test.cc | 650 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction() 658 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction() 766 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstructionWithCustomInit() 900 auto while_hlo1 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 902 auto while_hlo2 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 962 auto while_hlo1 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 964 auto while_hlo2 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 1262 auto while_hlo1 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 1264 auto while_hlo2 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 1332 HloInstruction::CreateWhile(loop_state_shape, condition, body, tuple)); in TEST_F() [all …]
|
D | hlo_alias_analysis_test.cc | 366 HloInstruction::CreateWhile(tuple_shape, condition, body, param)); in TEST_F() 557 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_F() 651 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_F() 653 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0)); in TEST_F() 655 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1)); in TEST_F() 737 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 750 HloInstruction::CreateWhile(tuple_shape, condition2, outer_body, tuple)); in TEST_F() 824 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_F() 974 HloInstruction::CreateWhile(tuple_shape, condition, body, select)); in TEST_F() 1068 HloInstruction::CreateWhile(init->shape(), condition, body, init)); in TEST_F()
|
D | flatten_call_graph_test.cc | 127 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 137 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 173 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
|
D | call_graph_test.cc | 326 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 336 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 460 a_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 471 entry_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
|
D | buffer_assignment_test.cc | 864 auto while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 1761 HloInstruction::CreateWhile(shape, condition, body, copy)); in TEST_F() 1941 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F() 1952 HloInstruction::CreateWhile(loop_state_shape_, cond1, body1, tuple1)); in TEST_F() 2169 HloInstruction::CreateWhile(r0s32, cond0, body0, infeed_data)); in TEST_F() 2174 HloInstruction::CreateWhile(r0s32, cond1, body1, while0)); in TEST_F() 2183 HloInstruction::CreateWhile(r0s32, cond2, body2, add)); in TEST_F() 2260 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F() 2268 HloInstruction::CreateWhile(loop_state_shape_, cond1, body1, while0)); in TEST_F() 2408 HloInstruction::CreateWhile(loop_state_shape_, cond, body, tuple0)); in TEST_F() [all …]
|
D | while_util.cc | 122 HloInstruction::CreateWhile(new_while_shape, new_while_condition, in MakeInstructionsLiveIn() 245 computation->AddInstruction(HloInstruction::CreateWhile( in MakeCountedLoop()
|
D | while_loop_simplifier.cc | 263 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in TryRemoveDeadWhileParams() 438 computation->AddInstruction(HloInstruction::CreateWhile( in TryRemoveConstantParams() 730 while_op, nested(computation->AddInstruction(HloInstruction::CreateWhile( in TryFlattenNestedTuples() 965 auto* new_while = computation->AddInstruction(HloInstruction::CreateWhile( in TryMergeInductionVariables()
|
D | call_inliner_test.cc | 105 HloInstruction::CreateWhile(pred, call_false, call_false, init_value)); in TEST_F()
|
D | hlo_ordering_test.cc | 151 HloInstruction::CreateWhile(scalar_shape, condition, body, constant)); in TEST_F() 234 HloInstruction::CreateWhile(scalar_shape, condition, body, constant)); in TEST_F()
|
D | bfloat16_propagation_test.cc | 600 HloInstruction::CreateWhile(shape, cond, body, add)); in TEST_F() 666 HloInstruction::CreateWhile(shape, cond, body, add)); in TEST_F() 743 HloInstruction::CreateWhile(tuple->shape(), cond, body, tuple)); in TEST_F() 865 HloInstruction::CreateWhile(tuple0->shape(), cond0, body, tuple0)); in TEST_F() 867 HloInstruction::CreateWhile(tuple1->shape(), cond1, body, tuple1)); in TEST_F()
|
D | hlo_dataflow_analysis_test.cc | 516 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P() 618 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P() 620 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0)); in TEST_P() 622 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1)); in TEST_P() 697 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P() 710 HloInstruction::CreateWhile(tuple_shape, condition, outer_body, tuple)); in TEST_P() 806 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P() 1066 HloInstruction::CreateWhile(tuple->shape(), condition, body, tuple)); in TEST_P() 1359 HloInstruction::CreateWhile(scalar_shape_, condition, body, param)); in TEST_P() 2617 auto whil = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
|
D | hlo_memory_scheduler_test.cc | 279 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
|
D | heap_simulator_test.cc | 80 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 172 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() 721 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
|
D | hlo_dce_test.cc | 249 auto live_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
|
D | hlo_rematerialization_test.cc | 116 auto while_inst = builder.AddInstruction(HloInstruction::CreateWhile( in MakeRematerializableWhileComputation()
|
D | hlo_instruction_test.cc | 1338 HloInstruction::CreateWhile(sout, computation, computation, x)); in TEST_F() 1541 HloInstruction::CreateWhile(sout, computation, computation, x)); in TEST_F() 1597 HloInstruction::CreateWhile(sout, computation, computation, x)); in TEST_F()
|
D | dynamic_dimension_inference_test.cc | 505 HloInstruction::CreateWhile(tuple_shape, condition, body, a_param)); in TEST_F()
|
D | hlo_instruction.h | 710 static std::unique_ptr<HloInstruction> CreateWhile(const Shape& shape,
|
D | tuple_points_to_analysis_test.cc | 1211 auto whil = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
|
D | hlo_instruction.cc | 940 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateWhile( in CreateWhile() function in xla::HloInstruction 1475 CreateWhile(shape, while_condition(), while_body(), new_operands[0]); in CloneWithNewOperands()
|
D | hlo_parser.cc | 953 instruction = builder->AddInstruction(HloInstruction::CreateWhile( in ParseInstructionRhs()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_copy_insertion_test.cc | 89 HloInstruction::CreateWhile(scalar_shape_, condition, body, param_0)); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | while_transformer_test.cc | 103 HloInstruction::CreateWhile(GetLoopStateShape(ind_var_tuple_index), in BuildWhileInstruction()
|