Searched refs:IsCustomCallToDnnBatchNorm (Results 1 – 3 of 3) sorted by relevance
48 if (!IsCustomCallToDnnBatchNorm(*hlo)) { in Run()
93 bool IsCustomCallToDnnBatchNorm(const HloInstruction& hlo);
116 bool IsCustomCallToDnnBatchNorm(const HloInstruction& hlo) { in IsCustomCallToDnnBatchNorm() function156 return ImplementedAsGemm(hlo) || IsCustomCallToDnnBatchNorm(hlo) || in ImplementedAsLibraryCall()