Searched refs:batch_norm_cpu_kernel (Results 1 – 1 of 1) sorted by relevance
1328 void batch_norm_cpu_kernel(Tensor& output, const Tensor& input, in batch_norm_cpu_kernel() function1399 REGISTER_DISPATCH(batch_norm_cpu_stub, &batch_norm_cpu_kernel);