Searched refs:_merge_dims_generic (Results 1 – 2 of 2) sorted by relevance
472 return _merge_dims_generic(current_field, new_parent_rank, inner_dim)1570 def _merge_dims_generic(source, outer, inner): function
1251 t2 = structured_tensor._merge_dims_generic(t, 1, 3)1262 t2 = structured_tensor._merge_dims_generic(t, 2, 2)