Searched refs:CreatePartitionId (Results 1 – 5 of 5) sorted by relevance
104 auto pid = comp->AddInstruction(HloInstruction::CreatePartitionId()); in DecomposeAllGather()116 shard_id = comp->AddInstruction(HloInstruction::CreatePartitionId()); in DecomposeAllGather()
713 static std::unique_ptr<HloInstruction> CreatePartitionId(
506 instruction = CreatePartitionId(shape); in CreateFromProto()1100 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreatePartitionId( in CreatePartitionId() function in xla::HloInstruction1820 clone = CreatePartitionId(shape); in CloneWithNewOperands()
1347 builder->AddInstruction(HloInstruction::CreatePartitionId()); in ParseInstructionRhs()
3415 return b->AddInstruction(HloInstruction::CreatePartitionId()); in GetDefaultCollectiveOpsCreator()