Searched defs:inputSizeDims (Results 1 – 1 of 1) sorted by relevance
181 inferDynamicDimsForConv(Location loc,Value input,Value weight,ShapedType resultTy,ArrayRef<int64_t> padAttr,ArrayRef<int64_t> strideAttr,ArrayRef<int64_t> dilationAttr,ArrayRef<int64_t> inputSizeDims,ArrayRef<int64_t> kernelSizeDims,OpBuilder & rewriter) inferDynamicDimsForConv() argument 265 llvm::SmallVector<int64_t> inputSizeDims; matchAndRewrite() local