Home
last modified time | relevance | path

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

12

/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.cc585 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction()
593 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction()
677 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstructionWithCustomInit()
770 while_hlo->parent()->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
783 outer_while_body->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
837 while_hlo->parent()->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
842 outer_while_body->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
896 while_hlo->parent()->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
909 outer_while_body->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
1007 auto while_hlo1 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
[all …]
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()
Dhlo_alias_analysis_test.cc393 HloInstruction::CreateWhile(tuple_shape, condition, body, param)); in TEST_F()
581 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_F()
673 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_F()
675 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0)); in TEST_F()
677 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1)); in TEST_F()
759 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
772 HloInstruction::CreateWhile(tuple_shape, condition2, outer_body, tuple)); in TEST_F()
846 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_F()
Dcall_graph_test.cc331 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
341 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
464 a_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
475 entry_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
Dhlo_rematerialization_test_utils.h115 auto while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
Dwhile_util.cc126 HloInstruction::CreateWhile(new_while_shape, new_while_condition, in MakeInstructionsLiveIn()
266 std::unique_ptr<HloInstruction> owned_while = HloInstruction::CreateWhile( in MakeCountedLoop()
Dbuffer_assignment_test.cc846 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
1055 auto while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
2167 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
2178 HloInstruction::CreateWhile(loop_state_shape_, cond1, body1, tuple1)); in TEST_F()
2395 HloInstruction::CreateWhile(r0s32, cond0, body0, infeed_data)); in TEST_F()
2400 HloInstruction::CreateWhile(r0s32, cond1, body1, while0)); in TEST_F()
2409 HloInstruction::CreateWhile(r0s32, cond2, body2, add)); in TEST_F()
2485 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
2493 HloInstruction::CreateWhile(loop_state_shape_, cond1, body1, while0)); in TEST_F()
2729 HloInstruction::CreateWhile(loop_state_shape_, cond, body, tuple0)); in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc430 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P()
532 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P()
534 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0)); in TEST_P()
536 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1)); in TEST_P()
598 auto level1_root = level1_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
607 auto level2_while = level2_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
622 HloInstruction::CreateWhile(tuple_shape, condition, level2_body, tuple)); in TEST_P()
705 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
718 HloInstruction::CreateWhile(tuple_shape, condition, outer_body, tuple)); in TEST_P()
811 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P()
[all …]
Dbfloat16_propagation_test.cc587 HloInstruction::CreateWhile(shape, cond, body, add)); in TEST_F()
653 HloInstruction::CreateWhile(shape, cond, body, add)); in TEST_F()
730 HloInstruction::CreateWhile(tuple->shape(), cond, body, tuple)); in TEST_F()
852 HloInstruction::CreateWhile(tuple0->shape(), cond0, body, tuple0)); in TEST_F()
854 HloInstruction::CreateWhile(tuple1->shape(), cond1, body, tuple1)); in TEST_F()
Dhlo_ordering_test.cc150 HloInstruction::CreateWhile(scalar_shape, condition, body, constant)); in TEST_F()
233 HloInstruction::CreateWhile(scalar_shape, condition, body, constant)); in TEST_F()
Dwhile_loop_simplifier.cc171 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in RemoveDeadTupleIndices()
754 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in TryRemoveConstantParams()
1085 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in TryFlattenNestedTuples()
1324 auto* new_while = computation->AddInstruction(HloInstruction::CreateWhile( in TryMergeInductionVariables()
Dcall_inliner_test.cc104 HloInstruction::CreateWhile(pred, call_false, call_false, init_value)); in TEST_F()
Dhlo_live_range_test.cc318 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
Dhlo_dce_test.cc287 auto live_while = 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()
914 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
Dwhile_loop_all_reduce_code_motion.cc708 while_instruction->parent()->AddInstruction(HloInstruction::CreateWhile( in AddSinkedAllReducesAndReplaceWhile()
Dhlo_instruction_test.cc1566 HloInstruction::CreateWhile(sout, computation, computation, x)); in TEST_F()
1846 HloInstruction::CreateWhile(sout, computation, computation, x)); in TEST_F()
1902 HloInstruction::CreateWhile(sout, computation, computation, x)); in TEST_F()
Dmemory_space_assignment_test.cc831 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
2642 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
2971 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
3072 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
3715 auto while0 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
Ddynamic_dimension_inference_test.cc780 HloInstruction::CreateWhile(tuple_shape, condition, body, a_param)); in TEST_F()
Dhlo_instruction.h1043 static std::unique_ptr<HloInstruction> CreateWhile(const Shape& shape,
Dhlo_instruction.cc1460 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateWhile( in CreateWhile() function in xla::HloInstruction
2130 CreateWhile(shape, while_condition(), while_body(), new_operands[0]); in CloneWithNewOperands()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc103 HloInstruction::CreateWhile(GetLoopStateShape(ind_var_tuple_index), in BuildWhileInstruction()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbuffer_donation_test.cc237 HloInstruction::CreateWhile(t_s32_f32v1_, condition, body, param)); in CreateTestModule()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc326 auto while_loop = b->AddInstruction(HloInstruction::CreateWhile( in GetFinalFftUsingCollectivePermute()

12