Lines Matching defs:weight

185     Location loc, Value input, Value weight, ShapedType resultTy,
205           rewriter.create<tensor::DimOp>(loc, weight, kernelDim);
247     Value weight = op->getOperand(1);
251     ShapedType weightTy = cast<ShapedType>(weight.getType());
265           op, "tosa.conv ops require static shapes for weight and bias");
279         loc, input, weight, resultTy, padAttr.asArrayRef(),
335         weight = rewriter.create<tosa::TransposeOp>(loc, newWeightTy, weight,
359       weight = rewriter.create<tosa::TransposeOp>(loc, newWeightTy, weight,
388                   loc, resultTy, ValueRange{input, weight, iZpVal, kZpVal},
398                          loc, resultTy, ValueRange{input, weight},
416     Value weight = op->getOperand(1);
420     ShapedType weightTy = cast<ShapedType>(weight.getType());
438         loc, input, weight, resultTy, padAttr.asArrayRef(),
518                            loc, linalgConvTy, ValueRange{input, weight},
547                   loc, linalgConvTy, ValueRange{input, weight, iZpVal, kZpVal},
632     auto weight = op.getWeight();
633     auto weightTy = cast<ShapedType>(weight.getType());
646       dynDims[1] = rewriter.create<tensor::DimOp>(loc, weight, 0);
661         loc, newWeightTy, weight, permutationValue);