Searched refs:_reshape_if_necessary (Results 1 – 1 of 1) sorted by relevance
387 t0 = _reshape_if_necessary(t0, new_shape)395 t1 = _reshape_if_necessary(t1, new_shape)403 product = _reshape_if_necessary(product, uncompacted_shape)420 def _reshape_if_necessary(tensor, new_shape): function