Searched refs:collapsed_dims (Results 1 – 2 of 2) sorted by relevance
608 linalg::ReassociationIndices collapsed_dims; in InsertReshapeIfNecessary() local614 collapsed_dims.push_back(index); in InsertReshapeIfNecessary()625 collapsed_dims_list.push_back(collapsed_dims); in InsertReshapeIfNecessary()626 collapsed_dims.clear(); in InsertReshapeIfNecessary()638 for (const auto& dims : collapsed_dims) { in InsertReshapeIfNecessary()
1640 SmallVector<int64_t, 2> collapsed_dims( in matchAndRewrite() local1651 /*collapsed_slice_dims=*/GetI64ElementsAttr(collapsed_dims, &rewriter), in matchAndRewrite()