Lines Matching refs:idx
688 for (int32_t idx = 0; idx < numInputDims; ++idx) {
689 if (getSizeOfDimension(input, idx) == 1) {
690 shouldSqueeze[idx] = true;
695 for (int32_t idx = 0; idx < squeezeDimsSize; ++idx) {
696 int32_t current = squeezeDims[idx] < 0 ? squeezeDims[idx] + numInputDims
697 : squeezeDims[idx];
735 for (int32_t idx = 0; idx < static_cast<int32_t>(numInputDims); ++idx) {
736 NN_OPS_CHECK(permData[idx] >= 0 && permData[idx] < static_cast<int32_t>(numInputDims));
737 outDims[idx] = getSizeOfDimension(input, permData[idx]);
764 for (int32_t idx = 0; idx < numInputDims; ++idx) {
767 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) {
773 outDims[idx] = 1;
775 outDims[idx] = getSizeOfDimension(input, idx);
802 for (int32_t idx = 0; idx < numInputDims; ++idx) {
805 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) {
812 outDims[idx - numSkipAxis] = getSizeOfDimension(input, idx);
849 for (int32_t idx = 0; idx < static_cast<int32_t>(numInputDims); idx++) {
850 int32_t dim = static_cast<int32_t>(getSizeOfDimension(input, idx));
851 int32_t stride = stridesData[idx];
856 int32_t begin = beginMask & (1 << idx)
858 : ClampedIndex(beginData[idx], dim, positiveStride);
859 int32_t end = endMask & (1 << idx)
861 : ClampedIndex(endData[idx], dim, positiveStride);
866 if (!(shrinkAxisMask & (1 << idx))) {
870 LOG(ERROR) << "Outdim " << idx << " is " << outDim << ", expected 1";