Searched refs:numpy_cube (Results 1 – 2 of 2) sorted by relevance
33 def numpy_cube(x: Tensor) -> Tuple[Tensor, Tensor]: function38 @numpy_cube.register_fake52 numpy_cube.register_autograd(numpy_cube_backward, setup_context=numpy_cube_setup_context)55 result = numpy_cube(x)58 numpy_cube.register_vmap(numpy_cube_vmap)447 op=numpy_cube._opoverload,
3846 torch.library.opcheck(custom_op_db.numpy_cube, args)