Searched refs:newWeightShape (Results 1 – 1 of 1) sorted by relevance
232 SmallVector<int64_t> newWeightShape{weightShape[1], weightShape[2], in matchAndRewrite() local239 RankedTensorType::get(newWeightShape, weightTy.getElementType()); in matchAndRewrite()617 SmallVector<int64_t> newWeightShape{weightShape[1], weightShape[0]}; in matchAndRewrite() local619 RankedTensorType::get(newWeightShape, weightTy.getElementType()); in matchAndRewrite()