Searched defs:inputDim (Results 1 – 3 of 3) sorted by relevance
152 getConvOrPoolOutputDim(Location loc,Value inputDim,int64_t padBeforeAttr,int64_t padAfterAttr,Value kernelDim,int64_t strideAttr,int64_t dilationAttr,OpBuilder & rewriter) getConvOrPoolOutputDim() argument 190 int64_t inputDim = inputSizeDims[i]; inferDynamicDimsForConv() local
1826 int64_t inputDim = inputDims[permutationRef[i]]; verify() local
3483 auto inputDim = type.getShape()[idx]; verify() local