Lines Matching defs:weightTy

41     ShapedType weightTy = cast<ShapedType>(weight.getType());
54     if (!inputTy.hasStaticShape() || !weightTy.hasStaticShape() ||
58     int64_t kernelHeight = weightTy.getDimSize(1);
59     int64_t kernelWidth = weightTy.getDimSize(2);
68         loc, weightTy, weight, /* axis = */ rewriter.getI32IntegerAttr(1));
70         loc, weightTy, reverse1, /* axis = */ rewriter.getI32IntegerAttr(2));
106     ShapedType weightTy = cast<ShapedType>(weight.getType());
111     Type weightETy = weightTy.getElementType();
126     if (!inputTy.hasStaticShape() || !weightTy.hasStaticShape() ||
132     int64_t outputChannels = weightTy.getDimSize(0);
133     int64_t weightHeight = weightTy.getDimSize(1);
134     int64_t weightWidth = weightTy.getDimSize(2);
135     int64_t inputChannels = weightTy.getDimSize(3);
160     weightTy = cast<ShapedType>(weight.getType());
161     weightHeight = weightTy.getDimSize(1);
162     weightWidth = weightTy.getDimSize(2);