Lines Matching refs:idx
65 for (int32_t idx = numInputDims - 1; idx >= 0; --idx) { in compute() local
66 starts.emplace_back(beginData[idx]); in compute()
67 stops.emplace_back(endData[idx]); in compute()
68 strides.emplace_back(stridesData[idx]); in compute()
170 for (int32_t idx = 0; idx < static_cast<int32_t>(numInputDims); idx++) { in prepare() local
171 int32_t dim = static_cast<int32_t>(getSizeOfDimension(inputShape, idx)); in prepare()
172 int32_t stride = stridesData[idx]; in prepare()
177 int32_t begin = beginMask & (1 << idx) ? positiveStride ? 0 : dim - 1 in prepare()
178 : ClampedIndex(beginData[idx], dim, positiveStride); in prepare()
179 int32_t end = endMask & (1 << idx) ? positiveStride ? dim : -1 in prepare()
180 : ClampedIndex(endData[idx], dim, positiveStride); in prepare()
185 if (!(shrinkAxisMask & (1 << idx))) { in prepare()
189 NN_RET_CHECK_GT(stride, 0) << "index = " << idx; in prepare()
190 NN_RET_CHECK_EQ(outDim, 1) << "index = " << idx; in prepare()