Searched refs:numpy_mul_setup_context (Results 1 – 1 of 1) sorted by relevance
69 def numpy_mul_setup_context(ctx, inputs, output): function78 numpy_mul.register_autograd(numpy_mul_backward, setup_context=numpy_mul_setup_context)