Searched refs:IsMatrixMultiplication (Results 1 – 7 of 7) sorted by relevance
87 if (!(IsCublasGemm(hlo) || IsMatrixMultiplication(hlo))) { in ComputeStreamToAssign()155 if (IsCublasGemm(*hlo) || IsMatrixMultiplication(*hlo)) { in AssignStreams()
50 if (IsMatrixMultiplication(*instr)) { in HandleDot()
70 bool IsMatrixMultiplication(const HloInstruction& dot);
154 if (IsMatrixMultiplication(*instr)) { in GetRelevantDots()
215 if (IsMatrixMultiplication(*instruction) && in AddBackendConstraints()
263 return IsMatrixMultiplication(dot) in OptimizeHloModule()446 return IsMatrixMultiplication(dot) ? candidate_operands in OptimizeHloPostLayoutAssignment()
106 bool IsMatrixMultiplication(const HloInstruction& dot) { in IsMatrixMultiplication() function