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.cc656 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction()
664 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction()
772 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstructionWithCustomInit()
866 while_hlo->parent()->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
879 outer_while_body->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
934 while_hlo->parent()->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
939 outer_while_body->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
994 while_hlo->parent()->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
1007 outer_while_body->AddInstruction(HloInstruction::CreateWhile( in TEST_F()
1105 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.cc360 HloInstruction::CreateWhile(tuple_shape, condition, body, param)); in TEST_F()
549 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_F()
643 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_F()
645 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0)); in TEST_F()
647 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1)); in TEST_F()
729 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
742 HloInstruction::CreateWhile(tuple_shape, condition2, outer_body, tuple)); in TEST_F()
816 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_F()
966 HloInstruction::CreateWhile(tuple_shape, condition, body, select)); in TEST_F()
1220 HloInstruction::CreateWhile(init->shape(), condition, body, init)); 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.cc827 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
1036 auto while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
2089 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
2100 HloInstruction::CreateWhile(loop_state_shape_, cond1, body1, tuple1)); in TEST_F()
2317 HloInstruction::CreateWhile(r0s32, cond0, body0, infeed_data)); in TEST_F()
2322 HloInstruction::CreateWhile(r0s32, cond1, body1, while0)); in TEST_F()
2331 HloInstruction::CreateWhile(r0s32, cond2, body2, add)); in TEST_F()
2407 HloInstruction::CreateWhile(loop_state_shape_, cond0, body0, tuple0)); in TEST_F()
2415 HloInstruction::CreateWhile(loop_state_shape_, cond1, body1, while0)); in TEST_F()
2590 HloInstruction::CreateWhile(loop_state_shape_, cond, body, tuple0)); in TEST_F()
[all …]
Dhlo_rematerialization_test_utils.h115 auto while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
Dwhile_util.cc122 HloInstruction::CreateWhile(new_while_shape, new_while_condition, in MakeInstructionsLiveIn()
260 computation->AddInstruction(HloInstruction::CreateWhile( in MakeCountedLoop()
Dhlo_dataflow_analysis_test.cc423 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P()
525 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P()
527 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0)); in TEST_P()
529 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1)); in TEST_P()
591 auto level1_root = level1_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
600 auto level2_while = level2_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
615 HloInstruction::CreateWhile(tuple_shape, condition, level2_body, tuple)); in TEST_P()
698 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
711 HloInstruction::CreateWhile(tuple_shape, condition, outer_body, tuple)); in TEST_P()
804 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P()
[all …]
Dwhile_loop_simplifier.cc160 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in RemoveDeadTupleIndices()
619 computation->AddInstruction(HloInstruction::CreateWhile( in TryRemoveConstantParams()
942 while_op, nested(computation->AddInstruction(HloInstruction::CreateWhile( in TryFlattenNestedTuples()
1177 auto* new_while = computation->AddInstruction(HloInstruction::CreateWhile( in TryMergeInductionVariables()
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()
Dbfloat16_propagation_test.cc630 HloInstruction::CreateWhile(shape, cond, body, add)); in TEST_F()
696 HloInstruction::CreateWhile(shape, cond, body, add)); in TEST_F()
773 HloInstruction::CreateWhile(tuple->shape(), cond, body, tuple)); in TEST_F()
895 HloInstruction::CreateWhile(tuple0->shape(), cond0, body, tuple0)); in TEST_F()
897 HloInstruction::CreateWhile(tuple1->shape(), cond1, body, tuple1)); in TEST_F()
Dcall_inliner_test.cc105 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()
Dwhile_loop_all_reduce_code_motion.cc500 while_instruction->parent()->AddInstruction(HloInstruction::CreateWhile( in AddSinkedAllReducesAndReplaceWhile()
Dhlo_dce_test.cc289 auto live_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
Dheap_simulator_test.cc81 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
173 builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
914 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()
Dhlo_instruction_test.cc1467 HloInstruction::CreateWhile(sout, computation, computation, x)); in TEST_F()
1674 HloInstruction::CreateWhile(sout, computation, computation, x)); in TEST_F()
1730 HloInstruction::CreateWhile(sout, computation, computation, x)); in TEST_F()
Dmemory_space_assignment_test.cc796 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
2681 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
3010 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
3111 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
3754 auto while0 = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_P()
Dhlo_instruction.h928 static std::unique_ptr<HloInstruction> CreateWhile(const Shape& shape,
Ddynamic_dimension_inference_test.cc773 HloInstruction::CreateWhile(tuple_shape, condition, body, a_param)); in TEST_F()
Dhlo_instruction.cc1186 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateWhile( in CreateWhile() function in xla::HloInstruction
1795 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.cc234 HloInstruction::CreateWhile(t_s32_f32v1_, condition, body, param)); in CreateTestModule()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc328 auto while_loop = b->AddInstruction(HloInstruction::CreateWhile( in GetFinalFftUsingCollectivePermute()

12