Searched refs:remainDimSize (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mnn/MNN-1.2.0/source/geometry/ |
| H A D | ConvertUtils.cpp | 143 int remainDimSize = sepInputShapeSize > 3 ? (int)sepInputShapeSize - 3 : 0; in broadcastto() local
144 std::vector<int> remainStride(remainDimSize + 1); in broadcastto()
145 … int remainSize = OpCommonUtils::computeStride(remainStride.data(), sepOutputShape, remainDimSize); in broadcastto()
147 std::vector<int> cords(remainDimSize + 1); in broadcastto()
149 OpCommonUtils::unravelIndexHelper(cords, remainStride, remainDimSize, index); in broadcastto()
151 for (int i = 0; i < remainDimSize; ++i) { in broadcastto()
|
| H A D | GeometryCrop.cpp | 48 int remainDimSize = seperateOffsets.size() > 3 ? (int)seperateOffsets.size() - 3 : 0; in computeOffsetRegion() local
49 remainStride.resize(remainDimSize); in computeOffsetRegion()
50 …Size = OpCommonUtils::computeStride(remainStride.data(), seperateOutputDims.data(), remainDimSize); in computeOffsetRegion()
54 std::vector<int> cords(remainDimSize); in computeOffsetRegion()
56 OpCommonUtils::unravelIndexHelper(cords, remainStride, remainDimSize, index); in computeOffsetRegion()
60 for (int i = 0; i < remainDimSize; ++i) { in computeOffsetRegion()
65 for (int i = remainDimSize; i < seperateOffsets.size(); ++i) { in computeOffsetRegion()
|