Searched refs:numpy_cat (Results 1 – 1 of 1) sorted by relevance
264 def numpy_cat(xs: Sequence[Tensor], dim: int) -> Tensor: function272 @numpy_cat.register_fake292 numpy_cat.register_autograd(numpy_cat_backward, setup_context=numpy_cat_setup_context)296 result = numpy_cat(x, dim)299 numpy_cat.register_vmap(numpy_cat_vmap)323 result = torch.ops._torch_testing.numpy_cat(grad_out, dim=ctx.dim)356 return torch.ops._torch_testing.numpy_cat(grad_out, dim=ctx.dim), None, None506 op=torch.ops._torch_testing.numpy_cat,