Searched refs:numpy_split_copy_setup_context (Results 1 – 1 of 1) sorted by relevance
318 def numpy_split_copy_setup_context(ctx, inputs, output): function326 …lit_copy.register_autograd(numpy_split_copy_backward, setup_context=numpy_split_copy_setup_context)