Searched defs:FusedBatchNormInferenceKernel (Results 1 – 1 of 1) sorted by relevance
130 struct FusedBatchNormInferenceKernel { struct134 __device__ static void run(int32 count, int32 channels_size, in run()178 struct FusedBatchNormInferenceKernel<Eigen::half, float, tensor_format, struct182 using IT = __half;186 using T = Eigen::half;187 using U = float;191 using GenericKernel =196 __device__ static void run(int32 count, int32 channels_size, in run()