Searched refs:ComputeAdam (Results 1 – 1 of 1) sorted by relevance
28 void ComputeAdam(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeAdam() function190 MultiThreadCompute<T>(ComputeAdam<T>, &input_params, unique_sparse_grad.indices_size_); in LaunchKernel()