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