Searched refs:numpy_sort_backward (Results 1 – 1 of 1) sorted by relevance
139 def numpy_sort_backward(ctx, grad_out, grad_ind, grad_ind_inv): function143 numpy_sort.register_autograd(numpy_sort_backward, setup_context=numpy_sort_setup_context)