Searched refs:numpy_sort_setup_context (Results 1 – 1 of 1) sorted by relevance
133 def numpy_sort_setup_context(ctx, inputs, output): function143 numpy_sort.register_autograd(numpy_sort_backward, setup_context=numpy_sort_setup_context)