Lines Matching refs:numInputDims
64 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); in compute() local
65 for (int32_t idx = numInputDims - 1; idx >= 0; --idx) { in compute()
71 for (int i = numInputDims; i < kMaxDim; i++) { in compute()
77 beginMask = ReverseMaskBits(beginMask, numInputDims); in compute()
78 endMask = ReverseMaskBits(endMask, numInputDims); in compute()
79 shrinkAxisMask = ReverseMaskBits(shrinkAxisMask, numInputDims); in compute()
141 uint32_t numInputDims = getNumberOfDimensions(inputShape); in prepare() local
142 NN_OPS_CHECK(numInputDims <= 4); in prepare()
152 NN_OPS_CHECK(getSizeOfDimension(beginShape, 0) == numInputDims); in prepare()
153 NN_OPS_CHECK(getSizeOfDimension(endShape, 0) == numInputDims); in prepare()
154 NN_OPS_CHECK(getSizeOfDimension(stridesShape, 0) == numInputDims); in prepare()
170 for (int32_t idx = 0; idx < static_cast<int32_t>(numInputDims); idx++) { in prepare()