Searched defs:FusedBatchNormOutputKernel (Results 1 – 1 of 1) sorted by relevance
186 struct FusedBatchNormOutputKernel { struct187 FusedBatchNormOutputKernel(T epsilon, const T* scaling_factor_data, in FusedBatchNormOutputKernel() function195 EIGEN_ALWAYS_INLINE void operator()( in operator()235 using WithFusedBatchNorm = FusedBatchNormOutputKernel<T>; argument