Home
last modified time | relevance | path

Searched refs:CreateWhile (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_invariant_code_motion_test.cc86 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()
Dcopy_insertion_test.cc650 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 …]
Dhlo_alias_analysis_test.cc366 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()
Dflatten_call_graph_test.cc127 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
137 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
173 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
Dcall_graph_test.cc326 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()
Dbuffer_assignment_test.cc864 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 …]
Dwhile_util.cc122 HloInstruction::CreateWhile(new_while_shape, new_while_condition, in MakeInstructionsLiveIn()
245 computation->AddInstruction(HloInstruction::CreateWhile( in MakeCountedLoop()
Dwhile_loop_simplifier.cc263 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()
Dcall_inliner_test.cc105 HloInstruction::CreateWhile(pred, call_false, call_false, init_value)); in TEST_F()
Dhlo_ordering_test.cc151 HloInstruction::CreateWhile(scalar_shape, condition, body, constant)); in TEST_F()
234 HloInstruction::CreateWhile(scalar_shape, condition, body, constant)); in TEST_F()
Dbfloat16_propagation_test.cc600 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()
Dhlo_dataflow_analysis_test.cc516 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()
Dhlo_memory_scheduler_test.cc279 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
Dheap_simulator_test.cc80 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()
Dhlo_dce_test.cc249 auto live_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
Dhlo_rematerialization_test.cc116 auto while_inst = builder.AddInstruction(HloInstruction::CreateWhile( in MakeRematerializableWhileComputation()
Dhlo_instruction_test.cc1338 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()
Ddynamic_dimension_inference_test.cc505 HloInstruction::CreateWhile(tuple_shape, condition, body, a_param)); in TEST_F()
Dhlo_instruction.h710 static std::unique_ptr<HloInstruction> CreateWhile(const Shape& shape,
Dtuple_points_to_analysis_test.cc1211 auto whil = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
Dhlo_instruction.cc940 /* 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()
Dhlo_parser.cc953 instruction = builder->AddInstruction(HloInstruction::CreateWhile( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_copy_insertion_test.cc89 HloInstruction::CreateWhile(scalar_shape_, condition, body, param_0)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc103 HloInstruction::CreateWhile(GetLoopStateShape(ind_var_tuple_index), in BuildWhileInstruction()