Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py2038 return torch.nested.nested_tensor_from_jagged(values * 2, offsets) * 2
2052 nt1 = torch.nested.nested_tensor_from_jagged(values * 2, offsets1)
2053 nt2 = torch.nested.nested_tensor_from_jagged(values * 3, offsets2)
2080 nt1 = torch.nested.nested_tensor_from_jagged(values * 2, offsets)
2081 nt2 = torch.nested.nested_tensor_from_jagged(values * 3, offsets2)
2089 nt2 = torch.nested.nested_tensor_from_jagged(other_values, nt.offsets())
2097 nt = torch.nested.nested_tensor_from_jagged(values, offsets)
2109 nt = torch.nested.nested_tensor_from_jagged(values, different_offsets)
2117 nt = torch.nested.nested_tensor_from_jagged(values, lengths=lengths)
2131 nt = torch.nested.nested_tensor_from_jagged(values, offsets)
[all …]
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/
H A Dnested.py70 yield torch.nested.nested_tensor_from_jagged(values, offsets)
222 njt = torch.nested.nested_tensor_from_jagged(
/aosp_15_r20/external/pytorch/test/
H A Dtest_flop_counter.py526 q = torch.nested.nested_tensor_from_jagged(q_values, q_offsets)
527 k = torch.nested.nested_tensor_from_jagged(k_values, k_offsets)
528 v = torch.nested.nested_tensor_from_jagged(v_values, k_offsets)
H A Dtest_nestedtensor.py281 nt = torch.nested.nested_tensor_from_jagged(
4011 nt = torch.nested.nested_tensor_from_jagged(values, offsets)
4937 nt_with_holes = torch.nested.nested_tensor_from_jagged(
4982 nt_with_holes = torch.nested.nested_tensor_from_jagged(
5031 nt_with_holes = torch.nested.nested_tensor_from_jagged(
5427 nt = torch.nested.nested_tensor_from_jagged(
5431 nt = torch.nested.nested_tensor_from_jagged(values, offsets=offsets)
5454 nt = torch.nested.nested_tensor_from_jagged(
5462 nt = torch.nested.nested_tensor_from_jagged(
5491 nt = torch.nested.nested_tensor_from_jagged(
[all …]
/aosp_15_r20/external/pytorch/torch/nested/
H A D__init__.py316 def nested_tensor_from_jagged( function
/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_execution_trace.py374 nt = torch.nested.nested_tensor_from_jagged(values, offsets)
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_codecache.py507 return torch.nested.nested_tensor_from_jagged(values, offsets)
H A Dtest_aot_inductor.py3220 nt = torch.nested.nested_tensor_from_jagged(values, offsets)