Searched refs:ComputeMomentum (Results 1 – 1 of 1) sorted by relevance
58 void ComputeMomentum(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeMomentum() function184 MultiThreadCompute<T>(ComputeMomentum<T>, &input_params, total_dim_size); in LaunchKernel()