ND Maximum and Minimum operators (with broadcasting support)
PiperOrigin-RevId: 284056046
diff --git a/test/binary-elementwise-operator-tester.h b/test/binary-elementwise-operator-tester.h
index 33c24b0..9a45854 100644
--- a/test/binary-elementwise-operator-tester.h
+++ b/test/binary-elementwise-operator-tester.h
@@ -26,6 +26,8 @@
enum class OperationType {
Unknown,
Add,
+ Maximum,
+ Minimum,
Multiply,
Subtract,
};
@@ -116,6 +118,10 @@
switch (operation_type()) {
case OperationType::Add:
return a + b;
+ case OperationType::Maximum:
+ return std::max<float>(a, b);
+ case OperationType::Minimum:
+ return std::min<float>(a, b);
case OperationType::Multiply:
return a * b;
case OperationType::Subtract:
@@ -211,6 +217,16 @@
output_min, output_max,
0, &binary_elementwise_op));
break;
+ case OperationType::Maximum:
+ ASSERT_EQ(xnn_status_success,
+ xnn_create_maximum_nd_f32(
+ 0, &binary_elementwise_op));
+ break;
+ case OperationType::Minimum:
+ ASSERT_EQ(xnn_status_success,
+ xnn_create_minimum_nd_f32(
+ 0, &binary_elementwise_op));
+ break;
case OperationType::Multiply:
ASSERT_EQ(xnn_status_success,
xnn_create_multiply_nd_f32(
@@ -243,6 +259,28 @@
input1.data(), input2.data(), output.data(),
nullptr /* thread pool */));
break;
+ case OperationType::Maximum:
+ ASSERT_EQ(xnn_status_success,
+ xnn_setup_maximum_nd_f32(
+ binary_elementwise_op,
+ num_input1_dims(),
+ input1_shape().data(),
+ num_input2_dims(),
+ input2_shape().data(),
+ input1.data(), input2.data(), output.data(),
+ nullptr /* thread pool */));
+ break;
+ case OperationType::Minimum:
+ ASSERT_EQ(xnn_status_success,
+ xnn_setup_minimum_nd_f32(
+ binary_elementwise_op,
+ num_input1_dims(),
+ input1_shape().data(),
+ num_input2_dims(),
+ input2_shape().data(),
+ input1.data(), input2.data(), output.data(),
+ nullptr /* thread pool */));
+ break;
case OperationType::Multiply:
ASSERT_EQ(xnn_status_success,
xnn_setup_multiply_nd_f32(
diff --git a/test/maximum-nd.cc b/test/maximum-nd.cc
new file mode 100644
index 0000000..b7cb07c
--- /dev/null
+++ b/test/maximum-nd.cc
@@ -0,0 +1,1154 @@
+// Copyright 2019 Google LLC
+//
+// This source code is licensed under the BSD-style license found in the
+// LICENSE file in the root directory of this source tree.
+
+#include <gtest/gtest.h>
+
+#include "binary-elementwise-operator-tester.h"
+
+constexpr size_t kDim1 = 2;
+constexpr size_t kDim2 = 3;
+constexpr size_t kDim3 = 4;
+constexpr size_t kDim4 = 5;
+constexpr size_t kDim5 = 6;
+constexpr size_t kDim6 = 7;
+
+
+TEST(MAXIMUM_ND_F32, 0d_x_0d) {
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .TestF32();
+}
+
+TEST(MAXIMUM_ND_F32, 1d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 0d_x_1d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 1d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 0d_x_2d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 1d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 2d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 2d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 2d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 0d_x_3d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 1d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 2d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 3d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 3d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 3d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 3d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 0d_x_4d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 1d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 2d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 3d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 4d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 4d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 4d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 4d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 4d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 0d_x_5d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 1d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 2d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 3d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 4d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 5d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 5d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 5d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 5d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 5d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 5d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .iterations(1)
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 0d_x_6d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 1d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 2d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 3d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 4d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 5d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .iterations(1)
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 6d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 6d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 6d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 6d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 6d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 6d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .iterations(1)
+ .TestF32();
+ }
+ }
+}
+
+TEST(MAXIMUM_ND_F32, 6d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Maximum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .iterations(1)
+ .TestF32();
+ }
+ }
+}
diff --git a/test/minimum-nd.cc b/test/minimum-nd.cc
new file mode 100644
index 0000000..edcc887
--- /dev/null
+++ b/test/minimum-nd.cc
@@ -0,0 +1,1154 @@
+// Copyright 2019 Google LLC
+//
+// This source code is licensed under the BSD-style license found in the
+// LICENSE file in the root directory of this source tree.
+
+#include <gtest/gtest.h>
+
+#include "binary-elementwise-operator-tester.h"
+
+constexpr size_t kDim1 = 2;
+constexpr size_t kDim2 = 3;
+constexpr size_t kDim3 = 4;
+constexpr size_t kDim4 = 5;
+constexpr size_t kDim5 = 6;
+constexpr size_t kDim6 = 7;
+
+
+TEST(MINIMUM_ND_F32, 0d_x_0d) {
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .TestF32();
+}
+
+TEST(MINIMUM_ND_F32, 1d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 0d_x_1d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 1d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 0d_x_2d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 1d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 2d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 2d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 2d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 0d_x_3d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 1d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 2d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 3d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 3d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 3d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 3d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 0d_x_4d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 1d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 2d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 3d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 4d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 4d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 4d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 4d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 4d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 0d_x_5d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 1d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 2d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 3d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 4d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 5d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 5d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 5d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 5d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 5d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 5d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .iterations(1)
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 0d_x_6d) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 1d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 1); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 2d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 2); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim2, input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 3d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 3); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 4d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 4); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 5d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 5); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .iterations(1)
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 6d_x_0d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .TestF32();
+ }
+}
+
+TEST(MINIMUM_ND_F32, 6d_x_1d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 1); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 6d_x_2d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 2); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 6d_x_3d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 3); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 6d_x_4d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 4); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 6d_x_5d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 5); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .iterations(1)
+ .TestF32();
+ }
+ }
+}
+
+TEST(MINIMUM_ND_F32, 6d_x_6d) {
+ for (uint32_t bm1 = 0; bm1 < (uint32_t(1) << 6); bm1++) {
+ for (uint32_t bm2 = 0; bm2 < (uint32_t(1) << 6); bm2++) {
+ const bool input1_broadcast_dim1 = bm1 & (uint32_t(1) << 0);
+ const bool input1_broadcast_dim2 = bm1 & (uint32_t(1) << 1);
+ const bool input1_broadcast_dim3 = bm1 & (uint32_t(1) << 2);
+ const bool input1_broadcast_dim4 = bm1 & (uint32_t(1) << 3);
+ const bool input1_broadcast_dim5 = bm1 & (uint32_t(1) << 4);
+ const bool input1_broadcast_dim6 = bm1 & (uint32_t(1) << 5);
+ const bool input2_broadcast_dim1 = bm2 & (uint32_t(1) << 0);
+ const bool input2_broadcast_dim2 = bm2 & (uint32_t(1) << 1);
+ const bool input2_broadcast_dim3 = bm2 & (uint32_t(1) << 2);
+ const bool input2_broadcast_dim4 = bm2 & (uint32_t(1) << 3);
+ const bool input2_broadcast_dim5 = bm2 & (uint32_t(1) << 4);
+ const bool input2_broadcast_dim6 = bm2 & (uint32_t(1) << 5);
+ const size_t input1_dim1 = input1_broadcast_dim1 ? 1 : kDim1;
+ const size_t input1_dim2 = input1_broadcast_dim2 ? 1 : kDim2;
+ const size_t input1_dim3 = input1_broadcast_dim3 ? 1 : kDim3;
+ const size_t input1_dim4 = input1_broadcast_dim4 ? 1 : kDim4;
+ const size_t input1_dim5 = input1_broadcast_dim5 ? 1 : kDim5;
+ const size_t input1_dim6 = input1_broadcast_dim6 ? 1 : kDim6;
+ const size_t input2_dim1 = input2_broadcast_dim1 ? 1 : kDim1;
+ const size_t input2_dim2 = input2_broadcast_dim2 ? 1 : kDim2;
+ const size_t input2_dim3 = input2_broadcast_dim3 ? 1 : kDim3;
+ const size_t input2_dim4 = input2_broadcast_dim4 ? 1 : kDim4;
+ const size_t input2_dim5 = input2_broadcast_dim5 ? 1 : kDim5;
+ const size_t input2_dim6 = input2_broadcast_dim6 ? 1 : kDim6;
+ BinaryElementwiseOperatorTester()
+ .operation_type(BinaryElementwiseOperatorTester::OperationType::Minimum)
+ .input1_shape({input1_dim6, input1_dim5, input1_dim4, input1_dim3, input1_dim2, input1_dim1})
+ .input2_shape({input2_dim6, input2_dim5, input2_dim4, input2_dim3, input2_dim2, input2_dim1})
+ .iterations(1)
+ .TestF32();
+ }
+ }
+}