Searched refs:numpy_take_setup_context (Results 1 – 1 of 1) sorted by relevance
169 def numpy_take_setup_context(ctx, inputs, output): function179 numpy_take.register_autograd(numpy_take_backward, setup_context=numpy_take_setup_context)