Searched refs:workspace_sparse_grad (Results 1 – 5 of 5) sorted by relevance
142 SparseGradient<T> workspace_sparse_grad({workspace_grad, workspace_indices, indices_size_}); in LaunchKernel() local146 param.workspace_grad_ = &workspace_sparse_grad; in LaunchKernel()
145 SparseGradient<T> workspace_sparse_grad({workspace_grad, workspace_indices, indices_size_}); in LaunchKernel() local149 param.workspace_grad_ = &workspace_sparse_grad; in LaunchKernel()
153 SparseGradient<T> workspace_sparse_grad({workspace_grad, workspace_indices, indices_size_}); in LaunchKernel() local157 param.workspace_grad_ = &workspace_sparse_grad; in LaunchKernel()
166 SparseGradient<T> workspace_sparse_grad({workspace_grad, workspace_indices, indices_size_}); in LaunchKernel() local170 param.workspace_grad_ = &workspace_sparse_grad; in LaunchKernel()
118 mindspore::kernel::SparseGradient<int> workspace_sparse_grad( in ReduceSparseGradient() local123 param.workspace_grad_ = &workspace_sparse_grad; in ReduceSparseGradient()