Searched refs:IsCublasGemm (Results 1 – 7 of 7) sorted by relevance
87 if (!(IsCublasGemm(hlo) || IsMatrixMultiplication(hlo))) { in ComputeStreamToAssign()155 if (IsCublasGemm(*hlo) || IsMatrixMultiplication(*hlo)) { in AssignStreams()
75 bool IsCublasGemm(const HloInstruction& hlo);
307 if (IsCublasGemm(*instr)) { in RunOnComputation()
209 CHECK(!IsCublasGemm(*instruction)) in AddBackendConstraints()
189 if (IsCublasGemm(*user)) { in CanShareBufferHint()
128 bool IsCublasGemm(const HloInstruction& hlo) { in IsCublasGemm() function212 return IsCublasGemm(hlo) || IsCustomCallToDnnBatchNorm(hlo) || in ImplementedAsLibraryCall()
623 if (xla::gpu::IsCublasGemm(*instr)) { in EmitCustomCallOp()