Home
last modified time | relevance | path

Searched refs:as_nested_tensor (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_nestedtensor.py272 nt = torch.nested.as_nested_tensor(values, layout=torch.jagged)
885 x = torch.nested.as_nested_tensor([a, b])
1328 nt_grad = torch.nested.as_nested_tensor([a, b])
2761 return torch.nested.as_nested_tensor(
2777 nt = torch.nested.as_nested_tensor([a, b])
2784 nt2 = torch.nested.as_nested_tensor([a, b])
2966 c = torch.nested.as_nested_tensor([a, b, c])
2999 nt0 = torch.nested.as_nested_tensor([a, b])
3000 nt1 = torch.nested.as_nested_tensor([c, d])
3037 nt0 = torch.nested.as_nested_tensor([a, b])
[all …]
H A Dtest_autograd.py5155 out = torch.nested.as_nested_tensor(
5162 grad = torch.nested.as_nested_tensor(
13881 nt = torch.nested.as_nested_tensor([a, b], dtype=torch.float, device=device)
13915 nt = torch.nested.as_nested_tensor([a, b], dtype=torch.float, device=device)
/aosp_15_r20/external/pytorch/torch/nested/
H A D__init__.py23 def as_nested_tensor( function
296 nt = as_nested_tensor(torch.unbind(tensor), layout=torch.strided).narrow(dim, start, length)
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/
H A Dnested.h73 inline at::Tensor as_nested_tensor(
/aosp_15_r20/external/pytorch/docs/source/
H A Dnested.rst58 In the vein of ``torch.as_tensor``, ``torch.nested.as_nested_tensor`` can be used to preserve autog…
174 .. autofunction:: as_nested_tensor
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/
H A Dnested.py111 return torch.nested.as_nested_tensor(out_ref_components, layout=torch.jagged)
/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dsdpa.py754 strided_nt = torch.nested.as_nested_tensor(list(tensor_list))
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py2276 nt = torch.nested.as_nested_tensor([a, b, c], layout=torch.jagged)