Searched refs:seperateOutputStrides (Results 1 – 2 of 2) sorted by relevance
43 seperateOutputStrides.resize(seperateOutputDims.size()); in computeOffsetRegion()62 reg.dst.offset += (cords[i] * seperateOutputStrides[i]); in computeOffsetRegion()75 reg.dst.stride[3 - i - 1] = seperateOutputStrides[match]; in computeOffsetRegion()110 std::vector<int> seperateOutputStrides; in onCompute() local145 std::vector<int> seperateOutputStrides; in onCompute() local243 dstBasicOffset += cords[i] * seperateOutputStrides[i]; in onCompute()269 dstOffset += seperateOffsets[si] * seperateOutputStrides[si]; in onCompute()272 region.dst.stride[di] = seperateOutputStrides[si]; in onCompute()280 region.dst.stride[di] = seperateOutputStrides[si]; in onCompute()290 region.dst.stride[di] = seperateOutputStrides[si]; in onCompute()[all …]
132 int seperateOutputStrides[MNN_MAX_TENSOR_DIM]; in broadcastto() local134 OpCommonUtils::computeStride(seperateOutputStrides, sepOutputShape, sepOutputShapeSize); in broadcastto()153 reg.dst.offset += (cords[i] * seperateOutputStrides[i]); in broadcastto()163 reg.dst.stride[3 - i - 1] = seperateOutputStrides[match]; in broadcastto()