Searched refs:weightDim (Results 1 – 1 of 1) sorted by relevance
108 int64_t weightDim = 2; in inferDynamicDimsForConv() local113 if (inputTy.isDynamicDim(i) && i != heightDim && i != weightDim) in inferDynamicDimsForConv()131 if (inputTy.isDynamicDim(weightDim)) { in inferDynamicDimsForConv()133 rewriter.create<tensor::DimOp>(loc, input, weightDim).getResult(); in inferDynamicDimsForConv()137 dynDims[weightDim] = getConvOutputDim( in inferDynamicDimsForConv()