Searched refs:numpy_cube_setup_context (Results 1 – 1 of 1) sorted by relevance
42 def numpy_cube_setup_context(ctx, inputs, output): function52 numpy_cube.register_autograd(numpy_cube_backward, setup_context=numpy_cube_setup_context)