Searched refs:convert_jagged_to_nested_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nestedtensor.py | 278 def convert_jagged_to_nested_tensor( function
6628 nt = convert_jagged_to_nested_tensor(values, offsets, max_length=4)
6650 nt32 = convert_jagged_to_nested_tensor(values32, offsets, max_length=16)
6651 nt16 = convert_jagged_to_nested_tensor(values16, offsets, max_length=16)
6741 nt = convert_jagged_to_nested_tensor(values, offsets, max_length=16)
6747 nt_meta = convert_jagged_to_nested_tensor(values, offsets, max_length=16)
6760 nt = convert_jagged_to_nested_tensor(values, offsets, max_length=4)
6774 nt = convert_jagged_to_nested_tensor(values, offsets, max_length=4)
6819 key = convert_jagged_to_nested_tensor(value, offsets, max_length_1)
6820 value = convert_jagged_to_nested_tensor(
|