Home
last modified time | relevance | path

Searched refs:startSx (Results 1 – 3 of 3) sorted by relevance

/dports/misc/mnn/MNN-1.2.0/source/geometry/
H A DGeometryConvUtils.cpp95 auto startSx = kx * dw - pw; in im2Col3d() local
97 if (startSx < 0) { in im2Col3d()
98 startDx = ((-startSx) + sw - 1) / sw; in im2Col3d()
99 startSx = startSx + startDx * sw; in im2Col3d()
107 if (startDx > endDx || endDx < 0 || startSx >= iw) { in im2Col3d()
174 auto startSx = kx * dw - pads.first; in im2Col() local
177 if (startSx < 0) { in im2Col()
178 startDx = ((-startSx) + sw - 1) / sw; in im2Col()
179 startSx = startSx + startDx * sw; in im2Col()
189 if (startDx > endDx || endDx < 0 || startSx >= iw) { in im2Col()
[all …]
H A DGeometryPoolGrad.cpp97 auto startSx = kx - pad_w; in onComputeMaxPool() local
99 if (startSx < 0) { in onComputeMaxPool()
100 startDx = ((-startSx) + stride_w - 1) / stride_w; in onComputeMaxPool()
101 startSx = startSx + startDx * stride_w; in onComputeMaxPool()
134 region.src.offset = startSy * iw + startSx; in onComputeMaxPool()
358 auto startSx = kx - pad_w; in onCompute() local
360 if (startSx < 0) { in onCompute()
361 startDx = ((-startSx) + stride_w - 1) / stride_w; in onCompute()
362 startSx = startSx + startDx * stride_w; in onCompute()
385 region.dst.offset = index * ob * oc * ih * iw + startSy * iw + startSx; in onCompute()
H A DGeometryConv2DBackPropFilter.cpp67 auto startSx = kx * dw - pads.first; in computeForDepthWise() local
69 if (startSx < 0) { in computeForDepthWise()
70 startDx = ((-startSx) + sw - 1) / sw; in computeForDepthWise()
71 startSx = startSx + startDx * sw; in computeForDepthWise()
83 auto srcOffsetKx = srcOffsetKy + startSx; in computeForDepthWise()