Lines Matching refs:weight

90     Value weight = op->getOperand(1);  in matchAndRewrite()  local
94 ShapedType weightTy = weight.getType().cast<ShapedType>(); in matchAndRewrite()
124 loc, weightTy, weight, rewriter.getI64IntegerAttr(1)); in matchAndRewrite()
154 Value weight = op->getOperand(1); in matchAndRewrite() local
158 ShapedType weightTy = weight.getType().cast<ShapedType>(); in matchAndRewrite()
205 weight = createOpAndInfer<tosa::PadOp>( in matchAndRewrite()
206 rewriter, loc, UnrankedTensorType::get(weightETy), weight, in matchAndRewrite()
211 weight = createOpAndInfer<tosa::PadOp>(rewriter, loc, in matchAndRewrite()
213 weight, weightPaddingVal); in matchAndRewrite()
216 weightTy = weight.getType().cast<ShapedType>(); in matchAndRewrite()
225 weight = createOpAndInfer<tosa::ReshapeOp>( in matchAndRewrite()
226 rewriter, loc, UnrankedTensorType::get(weightETy), weight, in matchAndRewrite()
234 weight = createOpAndInfer<tosa::TransposeOp>( in matchAndRewrite()
235 rewriter, loc, UnrankedTensorType::get(weightETy), weight, in matchAndRewrite()
242 weight = createOpAndInfer<tosa::ReshapeOp>( in matchAndRewrite()
243 rewriter, loc, UnrankedTensorType::get(weightETy), weight, in matchAndRewrite()
245 ShapedType restridedWeightTy = weight.getType().cast<ShapedType>(); in matchAndRewrite()
247 weight = createOpAndInfer<tosa::ReverseOp>( in matchAndRewrite()
248 rewriter, loc, UnrankedTensorType::get(weightETy), weight, in matchAndRewrite()
250 weight = createOpAndInfer<tosa::ReverseOp>( in matchAndRewrite()
251 rewriter, loc, UnrankedTensorType::get(weightETy), weight, in matchAndRewrite()
294 weight, zeroBias, in matchAndRewrite()
303 weight, zeroBias, in matchAndRewrite()