Searched refs:_transpose_first_two_dims (Results 1 – 1 of 1) sorted by relevance
3488 def _transpose_first_two_dims(value): function3530 value = _transpose_first_two_dims(value)3567 value = _transpose_first_two_dims(value)3862 elem = _transpose_first_two_dims(elem)3874 output = _transpose_first_two_dims(output)3893 output = _transpose_first_two_dims(output)3953 item = _transpose_first_two_dims(item)3962 tensor = _transpose_first_two_dims(tensor)