Home
last modified time | relevance | path

Searched refs:numpy_cube (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcustom_op_db.py33 def numpy_cube(x: Tensor) -> Tuple[Tensor, Tensor]: function
38 @numpy_cube.register_fake
52 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,
/aosp_15_r20/external/pytorch/test/
H A Dtest_custom_ops.py3846 torch.library.opcheck(custom_op_db.numpy_cube, args)