Lines Matching refs:getDimSize
114 int64_t kernelHeight = weightTy.getDimSize(1); in matchAndRewrite()
115 int64_t kernelWidth = weightTy.getDimSize(2); in matchAndRewrite()
185 int64_t batch = inputTy.getDimSize(0); in matchAndRewrite()
187 int64_t outputChannels = weightTy.getDimSize(0); in matchAndRewrite()
188 int64_t weightHeight = weightTy.getDimSize(1); in matchAndRewrite()
189 int64_t weightWidth = weightTy.getDimSize(2); in matchAndRewrite()
190 int64_t inputChannels = weightTy.getDimSize(3); in matchAndRewrite()
217 weightHeight = weightTy.getDimSize(1); in matchAndRewrite()
218 weightWidth = weightTy.getDimSize(2); in matchAndRewrite()
256 inputPadding[2] += restridedWeightTy.getDimSize(1) - 1; in matchAndRewrite()
257 inputPadding[3] += restridedWeightTy.getDimSize(1) - 1; in matchAndRewrite()
258 inputPadding[4] += restridedWeightTy.getDimSize(2) - 1; in matchAndRewrite()
259 inputPadding[5] += restridedWeightTy.getDimSize(2) - 1; in matchAndRewrite()
314 int64_t convHeight = convTy.getDimSize(1); in matchAndRewrite()
315 int64_t convWidth = convTy.getDimSize(2); in matchAndRewrite()