Lines Matching defs:weightTy

251     ShapedType weightTy = cast<ShapedType>(weight.getType());
263     if (!weightTy.hasStaticShape() || !biasTy.hasStaticShape())
283     auto weightShape = weightTy.getShape();
334             RankedTensorType::get(newWeightShape, weightTy.getElementType());
358           RankedTensorType::get(newWeightShape, weightTy.getElementType());
420     ShapedType weightTy = cast<ShapedType>(weight.getType());
432     if (!weightTy.hasStaticShape() || !biasTy.hasStaticShape())
453     auto weightShape = weightTy.getShape();
633     auto weightTy = cast<ShapedType>(weight.getType());
634     auto weightShape = weightTy.getShape();
645     if (!weightTy.hasRank() || weightTy.isDynamicDim(0)) {
658         RankedTensorType::get(newWeightShape, weightTy.getElementType());