Lines Matching defs:weight

37     Value weight = op->getOperand(1);
41     ShapedType weightTy = cast<ShapedType>(weight.getType());
68         loc, weightTy, weight, /* axis = */ rewriter.getI32IntegerAttr(1));
102     Value weight = op->getOperand(1);
106     ShapedType weightTy = cast<ShapedType>(weight.getType());
137     // Pad the weight so that it is modulo of the striding.
149       weight = CreateOpAndInferShape<tosa::PadOp>(
150           rewriter, loc, UnrankedTensorType::get(weightETy), weight,
155       weight = CreateOpAndInferShape<tosa::PadOp>(
156           rewriter, loc, UnrankedTensorType::get(weightETy), weight,
160     weightTy = cast<ShapedType>(weight.getType());
169     weight = CreateOpAndInferShape<tosa::ReshapeOp>(
170         rewriter, loc, UnrankedTensorType::get(weightETy), weight,
178     weight = CreateOpAndInferShape<tosa::TransposeOp>(
179         rewriter, loc, UnrankedTensorType::get(weightETy), weight,
186     weight = CreateOpAndInferShape<tosa::ReshapeOp>(
187         rewriter, loc, UnrankedTensorType::get(weightETy), weight,
189     ShapedType restridedWeightTy = cast<ShapedType>(weight.getType());
191     weight = CreateOpAndInferShape<tosa::ReverseOp>(
192         rewriter, loc, UnrankedTensorType::get(weightETy), weight,
194     weight = CreateOpAndInferShape<tosa::ReverseOp>(
195         rewriter, loc, UnrankedTensorType::get(weightETy), weight,
235                    weight, zeroBias,
244                    weight, zeroBias,