Searched refs:collapsed_dims_list (Results 1 – 1 of 1) sorted by relevance
607 SmallVector<linalg::ReassociationIndices, 4> collapsed_dims_list; in InsertReshapeIfNecessary() local625 collapsed_dims_list.push_back(collapsed_dims); in InsertReshapeIfNecessary()632 if (collapsed_dims_list.empty()) { in InsertReshapeIfNecessary()633 collapsed_dims_list.push_back({}); in InsertReshapeIfNecessary()639 collapsed_dims_list.back().push_back(dims); in InsertReshapeIfNecessary()651 collapsed_dims_list); in InsertReshapeIfNecessary()