Searched refs:a2_reshape_dims (Results 1 – 1 of 1) sorted by relevance
1292 SmallVector<int64_t, 4> a2_reshape_dims; in matchAndRewrite() local1293 a2_reshape_dims.push_back(a1_transpose_dims[0]); in matchAndRewrite()1294 a2_reshape_dims.push_back(a1_transpose_dims[1]); in matchAndRewrite()1295 a2_reshape_dims.push_back(a1_transpose_dims[2] / depth_multiplier.getInt()); in matchAndRewrite()1296 a2_reshape_dims.push_back(depth_multiplier.getInt()); in matchAndRewrite()1311 RankedTensorType::get(ArrayRef<int64_t>(a2_reshape_dims), in matchAndRewrite()1314 rewriter.getI64ArrayAttr(a2_reshape_dims)); in matchAndRewrite()