Searched defs:pgather (Results 1 – 1 of 1) sorted by relevance
151 auto pgather = b->AddInstruction(gather->CloneWithNewOperands( in PartitionIndexOnlyPartition() local199 auto pgather = b->AddInstruction(HloInstruction::CreateGather( in ParititonPassthroughOperand() local248 auto pgather = b->AddInstruction(HloInstruction::CreateGather( in ParititonTrivialIndexedOperandDimension() local424 HloInstruction* pgather; in PartitionIndexParallelDimensions() local683 HloInstruction* pgather; in HandleGather() local