Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py11285 def maybe_index_tuple(maybe_tuple_of_tensors, index): argument
11286 if maybe_tuple_of_tensors is None:
11288 return tuple(maybe_tuple_of_tensors[j][:, index:index + 1, :].contiguous()