ND elementwise Multiply operator with broadcasting support
PiperOrigin-RevId: 280801113
diff --git a/test/multiply.cc b/test/multiply.cc
new file mode 100644
index 0000000..2c7bb72
--- /dev/null
+++ b/test/multiply.cc
@@ -0,0 +1,125 @@
+// 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 "multiply-operator-tester.h"
+
+
+TEST(MULTIPLY_OP_F32, vector_x_vector) {
+ MultiplyOperatorTester()
+ .input1_shape({2})
+ .input2_shape({2})
+ .iterations(1)
+ .TestF32();
+}
+
+TEST(MULTIPLY_OP_F32, vector_x_scalar) {
+ MultiplyOperatorTester()
+ .input1_shape({2})
+ .iterations(1)
+ .TestF32();
+ MultiplyOperatorTester()
+ .input1_shape({2})
+ .input2_shape({1})
+ .iterations(1)
+ .TestF32();
+}
+
+TEST(MULTIPLY_OP_F32, scalar_x_vector) {
+ MultiplyOperatorTester()
+ .input2_shape({2})
+ .iterations(1)
+ .TestF32();
+ MultiplyOperatorTester()
+ .input1_shape({1})
+ .input2_shape({2})
+ .iterations(1)
+ .TestF32();
+}
+
+TEST(MULTIPLY_OP_F32, matrix_x_matrix) {
+ MultiplyOperatorTester()
+ .input1_shape({2, 3})
+ .input2_shape({2, 3})
+ .iterations(1)
+ .TestF32();
+}
+
+TEST(MULTIPLY_OP_F32, matrix_x_row) {
+ MultiplyOperatorTester()
+ .input1_shape({2, 3})
+ .input2_shape({1, 3})
+ .iterations(1)
+ .TestF32();
+ MultiplyOperatorTester()
+ .input1_shape({2, 3})
+ .input2_shape({3})
+ .iterations(1)
+ .TestF32();
+}
+
+TEST(MULTIPLY_OP_F32, matrix_x_column) {
+ MultiplyOperatorTester()
+ .input1_shape({2, 3})
+ .input2_shape({2, 1})
+ .iterations(1)
+ .TestF32();
+}
+
+TEST(MULTIPLY_OP_F32, matrix_x_scalar) {
+ MultiplyOperatorTester()
+ .input1_shape({2, 3})
+ .input2_shape({1, 1})
+ .iterations(1)
+ .TestF32();
+ MultiplyOperatorTester()
+ .input1_shape({2, 3})
+ .input2_shape({1})
+ .iterations(1)
+ .TestF32();
+ MultiplyOperatorTester()
+ .input1_shape({2, 3})
+ .iterations(1)
+ .TestF32();
+}
+
+TEST(MULTIPLY_OP_F32, row_x_matrix) {
+ MultiplyOperatorTester()
+ .input1_shape({1, 3})
+ .input2_shape({2, 3})
+ .iterations(1)
+ .TestF32();
+ MultiplyOperatorTester()
+ .input1_shape({3})
+ .input2_shape({2, 3})
+ .iterations(1)
+ .TestF32();
+}
+
+TEST(MULTIPLY_OP_F32, column_x_matrix) {
+ MultiplyOperatorTester()
+ .input1_shape({2, 1})
+ .input2_shape({2, 3})
+ .iterations(1)
+ .TestF32();
+}
+
+TEST(MULTIPLY_OP_F32, scalar_x_matrix) {
+ MultiplyOperatorTester()
+ .input1_shape({1, 1})
+ .input2_shape({2, 3})
+ .iterations(1)
+ .TestF32();
+ MultiplyOperatorTester()
+ .input1_shape({1})
+ .input2_shape({2, 3})
+ .iterations(1)
+ .TestF32();
+ MultiplyOperatorTester()
+ .input2_shape({2, 3})
+ .iterations(1)
+ .TestF32();
+}