Searched refs:_reshape_if_necessary (Results 1 – 1 of 1) sorted by relevance
1004 t0 = _reshape_if_necessary(t0, new_shape)1012 t1 = _reshape_if_necessary(t1, new_shape)1020 product = _reshape_if_necessary(product, uncompacted_shape)1037 def _reshape_if_necessary(tensor, new_shape): function