Searched refs:numpy_split_copy_with_int (Results 1 – 1 of 1) sorted by relevance
342 def numpy_split_copy_with_int(x: Tensor, splits: Sequence[int], dim: int) -> Tuple[List[Tensor], in… function347 @numpy_split_copy_with_int.register_fake358 numpy_split_copy_with_int.register_autograd(365 result, len_split = numpy_split_copy_with_int(x, splits, dim + 1)368 numpy_split_copy_with_int.register_vmap(numpy_split_copy_with_int_vmap)526 op=torch.ops._torch_testing.numpy_split_copy_with_int,