Searched refs:IsRank2Transpose (Results 1 – 7 of 7) sorted by relevance
46 if (operand.IsRank2Transpose()) { in CanFoldOperandsIntoDot()
1160 if (lhs->IsRank2Transpose()) { in HandleDotStrengthReduction()1172 if (rhs->IsRank2Transpose()) { in HandleDotStrengthReduction()1806 lhs->IsRank2Transpose() && rhs->IsRank2Transpose()) { in HandleDot()
1407 bool IsRank2Transpose() const;
536 bool IsRank2Transpose() const;
3242 bool HloInstruction::IsRank2Transpose() const { in IsRank2Transpose() function in xla::HloInstruction3244 return transpose != nullptr && transpose->IsRank2Transpose(); in IsRank2Transpose()
832 bool HloTransposeInstruction::IsRank2Transpose() const { in IsRank2Transpose() function in xla::HloTransposeInstruction
138 if (hlo.IsRank2Transpose()) { in StripTranspose()