Searched refs:IsCustomCallToDnnBatchNorm (Results 1 – 4 of 4) sorted by relevance
48 if (!IsCustomCallToDnnBatchNorm(*hlo)) { in Run()
105 bool IsCustomCallToDnnBatchNorm(const HloInstruction& hlo);
171 bool IsCustomCallToDnnBatchNorm(const HloInstruction& hlo) { in IsCustomCallToDnnBatchNorm() function212 return IsCublasGemm(hlo) || IsCustomCallToDnnBatchNorm(hlo) || in ImplementedAsLibraryCall()
631 if (xla::gpu::IsCustomCallToDnnBatchNorm(*instr)) { in EmitCustomCallOp()