Searched refs:CreateConditional (Results 1 – 16 of 16) sorted by relevance
241 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
95 builder.AddInstruction(HloInstruction::CreateConditional( in MakeConditional()
1668 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()
1048 static std::unique_ptr<HloInstruction> CreateConditional(1053 static std::unique_ptr<HloInstruction> CreateConditional(
268 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
317 CreateConditional(shape, cond_operands[0], all_computations(), in CreateFromProto()1472 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConditional( in CreateConditional() function in xla::HloInstruction1489 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConditional( in CreateConditional() function in xla::HloInstruction2134 clone = CreateConditional(shape, new_operands[0], in CloneWithNewOperands()
1574 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()1907 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
888 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
760 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
575 conditional_parent->AddInstruction(HloInstruction::CreateConditional( in ConvertSpecialMove()
1591 hlo->parent()->AddInstruction(HloInstruction::CreateConditional( in HandleConditional()
1120 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
2577 return builder->AddInstruction(HloInstruction::CreateConditional( in CreateInstruction()
2856 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
2884 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()
1728 conditional = body_b.AddInstruction(HloInstruction::CreateConditional( in PartitionBaseCase()