Searched refs:non_unit_dims (Results 1 – 1 of 1) sorted by relevance
82 SmallVector<int64_t, 4> non_unit_dims; in matchAndRewrite() local87 non_unit_dims.push_back(dim); in matchAndRewrite()95 if (reshape_shape.size() < non_unit_dims.size()) return failure(); in matchAndRewrite()112 if (non_unit_dims != old_reshape_non_unit_dims) return failure(); in matchAndRewrite()