Searched refs:non_feature_dims (Results 1 – 1 of 1) sorted by relevance
1856 SmallVector<int64_t, 4> non_feature_dims; in matchAndRewrite() local1859 non_feature_dims.push_back(i); in matchAndRewrite()1861 auto reduce_dims = GetI64ElementsAttr(non_feature_dims, &rewriter); in matchAndRewrite()