Searched refs:CreateConditional (Results 1 – 16 of 16) sorted by relevance
241 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
96 builder.AddInstruction(HloInstruction::CreateConditional( in MakeConditional()
933 static std::unique_ptr<HloInstruction> CreateConditional(938 static std::unique_ptr<HloInstruction> CreateConditional(
263 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
1740 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()1834 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()1949 inner_builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()1969 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
252 CreateConditional(shape, cond_operands[0], all_computations(), in CreateFromProto()1198 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConditional( in CreateConditional() function in xla::HloInstruction1215 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConditional( in CreateConditional() function in xla::HloInstruction1799 clone = CreateConditional(shape, new_operands[0], in CloneWithNewOperands()
1475 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()1735 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
489 conditional_parent->AddInstruction(HloInstruction::CreateConditional( in ConvertSpecialMove()
797 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
881 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
1373 hlo->parent()->AddInstruction(HloInstruction::CreateConditional( in HandleConditional()
1101 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
2230 instruction = builder->AddInstruction(HloInstruction::CreateConditional( in ParseInstructionRhs()
2895 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
2223 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleSingleDevice()2615 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleInfeed()2888 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleConditional()3034 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleOutfeed()
1376 conditional = body_b.AddInstruction(HloInstruction::CreateConditional( in PartitionBaseCase()