Home
last modified time | relevance | path

Searched refs:CreateConditional (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dflatten_call_graph_test.cc241 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
H A Dconditional_simplifier_test.cc95 builder.AddInstruction(HloInstruction::CreateConditional( in MakeConditional()
H A Dhlo_dataflow_analysis_test.cc1668 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
1762 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
1877 inner_builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
1897 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
H A Dhlo_instruction.h1048 static std::unique_ptr<HloInstruction> CreateConditional(
1053 static std::unique_ptr<HloInstruction> CreateConditional(
H A Dcall_graph_test.cc268 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
H A Dhlo_instruction.cc317 CreateConditional(shape, cond_operands[0], all_computations(), in CreateFromProto()
1472 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConditional( in CreateConditional() function in xla::HloInstruction
1489 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConditional( in CreateConditional() function in xla::HloInstruction
2134 clone = CreateConditional(shape, new_operands[0], in CloneWithNewOperands()
H A Dhlo_instruction_test.cc1574 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
1907 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
H A Ddynamic_dimension_inference_test.cc888 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
H A Dlayout_assignment_test.cc760 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
H A Dconditional_code_motion.cc575 conditional_parent->AddInstruction(HloInstruction::CreateConditional( in ConvertSpecialMove()
H A Ddynamic_dimension_inference.cc1591 hlo->parent()->AddInstruction(HloInstruction::CreateConditional( in HandleConditional()
H A Dbuffer_assignment_test.cc1120 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
H A Dhlo_parser.cc2577 return builder->AddInstruction(HloInstruction::CreateConditional( in CreateInstruction()
H A Dmemory_space_assignment_test.cc2856 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc2884 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleSingleDevice()
3371 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleInfeed()
3668 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleConditional()
3816 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleOutfeed()
H A Ddot_handler.cc1728 conditional = body_b.AddInstruction(HloInstruction::CreateConditional( in PartitionBaseCase()