Searched refs:CreateCollectivePermute (Results 1 – 3 of 3) sorted by relevance
515 static std::unique_ptr<HloInstruction> CreateCollectivePermute(
408 CreateCollectivePermute(shape, operands(0), source_target_pairs); in CreateFromProto()850 HloInstruction::CreateCollectivePermute( in CreateCollectivePermute() function in xla::HloInstruction
876 HloInstruction::CreateCollectivePermute(shape, operands[0], pairs)); in ParseInstructionRhs()