Lines Matching refs:idx
60 for (int32_t idx = numInputDims - 1; idx >= 0; --idx) { in compute() local
61 starts.emplace_back(beginData[idx]); in compute()
62 stops.emplace_back(endData[idx]); in compute()
63 strides.emplace_back(stridesData[idx]); in compute()
129 for (int32_t idx = 0; idx < static_cast<int32_t>(numInputDims); idx++) { in prepare() local
130 int32_t dim = static_cast<int32_t>(getSizeOfDimension(inputShape, idx)); in prepare()
131 int32_t stride = stridesData[idx]; in prepare()
136 int32_t begin = beginMask & (1 << idx) ? positiveStride ? 0 : dim - 1 in prepare()
137 : ClampedIndex(beginData[idx], dim, positiveStride); in prepare()
138 int32_t end = endMask & (1 << idx) ? positiveStride ? dim : -1 in prepare()
139 : ClampedIndex(endData[idx], dim, positiveStride); in prepare()
144 if (!(shrinkAxisMask & (1 << idx))) { in prepare()
148 NN_RET_CHECK_GT(stride, 0) << "index = " << idx; in prepare()
149 NN_RET_CHECK_EQ(outDim, 1) << "index = " << idx; in prepare()