chore: import upstream snapshot with attribution

2026-07-13 12:47:05 +08:00
commit 4f3b7da785
7394 changed files with 2005594 additions and 0 deletions
@@ -0,0 +1,84 @@
+/* ******************************************************************************
+ *
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Apache License, Version 2.0 which is available at
+ * https://www.apache.org/licenses/LICENSE-2.0.
+ *
+ *  See the NOTICE file distributed with this work for additional
+ *  information regarding copyright ownership.
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ * License for the specific language governing permissions and limitations
+ * under the License.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ ******************************************************************************/
+
+//
+//  @author raver119@gmail.com
+//
+#include <ops/declarable/helpers/convolutions.h>
+#include <ops/declarable/helpers/max_pooling.h>
+
+namespace sd {
+namespace ops {
+namespace helpers {
+
+template <typename T>
+static void maxPoolingFunctor_(sd::graph::Context& block, NDArray* input, NDArray* values,
+                               const std::vector<LongType>& params, NDArray* indices) {
+  LongType kY = params[0];
+  LongType kX = params[1];
+
+  LongType sY = params[2];
+  LongType sX = params[3];
+
+  sd::LongType  pY = params[4];
+  sd::LongType  pX = params[5];
+
+  LongType dY = params[6];
+  LongType dX = params[7];
+
+  LongType oY = 0;
+  LongType oX = 0;
+
+  const LongType bSize = input->sizeAt(0);
+  const LongType inD = input->sizeAt(1);
+  const LongType inY = input->sizeAt(2);
+  const LongType inX = input->sizeAt(3);
+
+  const bool isSameMode = params[8] != 0;
+
+  ConvolutionUtils::calcOutSizePool2D(oY, oX, kY, kX, sY, sX, pY, pX, dY, dX, inY, inX, isSameMode);
+
+  if (isSameMode)
+    ConvolutionUtils::calcPadding2D(pY, pX, oY, oX, inY, inX, params[0], params[1], params[2], params[3], params[6],
+                                    params[7]);
+
+  // 0,1 - kernel Height/Width; 2,3 - stride Height/Width; 4,5 - pad Height/Width; 6,7 - dilation Height/Width; 8 -
+  // poolingMode; 9 - divisor;
+  ConvolutionUtils::pooling2d(block, *input, *values, kY, kX, sY, sX, pY, pX, dY, dX, PoolingType::MAX_POOL, 1);
+
+  if (nullptr != indices) {
+    // for max_pool_with_argmax
+    int total = input->lengthOf();
+    int part = total / bSize;
+
+    for (int k = 0; k < total;)
+      for (int i = 0; i < part; i++) {
+        indices->p(k++, i);
+      }
+  }
+}
+
+void maxPoolingFunctor(sd::LaunchContext* context, sd::graph::Context& block, NDArray* input, NDArray* values,
+                       const std::vector<LongType>& params, NDArray* indices) {
+  BUILD_SINGLE_SELECTOR(input->dataType(), maxPoolingFunctor_, (block, input, values, params, indices),
+                        SD_COMMON_TYPES);
+}
+
+}  // namespace helpers
+}  // namespace ops
+}  // namespace sd