Searched refs:weightWDim (Results 1 – 1 of 1) sorted by relevance
103 int64_t weightHDim, int64_t weightWDim, OpBuilder &rewriter) { in inferDynamicDimsForConv() argument135 rewriter.create<tensor::DimOp>(loc, weight, weightWDim).getResult(); in inferDynamicDimsForConv()