Searched refs:maybe_tuple_of_tensors (Results 1 – 1 of 1) sorted by relevance
11285 def maybe_index_tuple(maybe_tuple_of_tensors, index): argument11286 if maybe_tuple_of_tensors is None:11288 return tuple(maybe_tuple_of_tensors[j][:, index:index + 1, :].contiguous()