Searched defs:FusedBatchNormInferenceKernel (Results 1 – 1 of 1) sorted by relevance
128 struct FusedBatchNormInferenceKernel { struct132 __device__ static void run(int32 count, int32 channels_size, in run()176 struct FusedBatchNormInferenceKernel<Eigen::half, float, tensor_format, struct180 using IT = __half;184 using T = Eigen::half;185 using U = float;189 using GenericKernel =194 __device__ static void run(int32 count, int32 channels_size, in run()