Searched refs:ImplementedAsGemm (Results 1 – 7 of 7) sorted by relevance
82 if (!ImplementedAsGemm(hlo)) { in ComputeStreamToAssign()150 if (ImplementedAsGemm(*hlo)) { in AssignStreams()
68 bool ImplementedAsGemm(const HloInstruction& hlo);
86 bool ImplementedAsGemm(const HloInstruction& hlo) { in ImplementedAsGemm() function156 return ImplementedAsGemm(hlo) || IsCustomCallToDnnBatchNorm(hlo) || in ImplementedAsLibraryCall()
171 ImplementedAsGemm(*consumer)) { in ShouldFuseInexpensiveChecks()
191 ImplementedAsGemm(*instruction) && in AddBackendConstraints()
247 return ImplementedAsGemm(dot) ? candidate_operands in OptimizeHloModule()
353 if (ImplementedAsGemm(*dot)) { in HandleDot()675 if (ImplementedAsGemm(*fusion)) { in HandleFusion()