summaryrefslogtreecommitdiff
path: root/nn/runtime/test/specs/V1_3/div_int32.mod.py
diff options
context:
space:
mode:
authorSlava Shklyaev <slavash@google.com>2020-01-09 15:16:50 +0000
committerSlava Shklyaev <slavash@google.com>2020-01-13 15:44:27 +0000
commitb88bb3e3f225d5262499f5f072fcfb20d0e59951 (patch)
tree39041e42a817946d94c02a2d84394a4deb8cc337 /nn/runtime/test/specs/V1_3/div_int32.mod.py
parentc01d2a24ac6de9e9db8a503bb54d37abc5feafd7 (diff)
downloadml-b88bb3e3f225d5262499f5f072fcfb20d0e59951.tar.gz
Add TENSOR_INT32 support for DIV
Bug: 142318767 Bug: 136735929 Test: NNT_static Change-Id: I105222e43802bc58562bb904bd9e9b306eb3697c
Diffstat (limited to 'nn/runtime/test/specs/V1_3/div_int32.mod.py')
-rw-r--r--nn/runtime/test/specs/V1_3/div_int32.mod.py76
1 files changed, 76 insertions, 0 deletions
diff --git a/nn/runtime/test/specs/V1_3/div_int32.mod.py b/nn/runtime/test/specs/V1_3/div_int32.mod.py
new file mode 100644
index 000000000..d378f5973
--- /dev/null
+++ b/nn/runtime/test/specs/V1_3/div_int32.mod.py
@@ -0,0 +1,76 @@
+#
+# Copyright (C) 2020 The Android Open Source Project
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+model = Model()
+input0 = Input("input0", "TENSOR_INT32", "{2, 2, 4, 6}")
+input1 = Input("input1", "TENSOR_INT32", "{2, 2, 4, 6}")
+output = Output("output", "TENSOR_INT32", "{2, 2, 4, 6}")
+model = model.Operation("DIV", input0, input1, 0).To(output)
+Example({
+ input0: [
+ -6, -6, -6, -6, -6, -6,
+ -5, -5, -5, -5, -5, -5,
+ -4, -4, -4, -4, -4, -4,
+ -3, -3, -3, -3, -3, -3,
+ -2, -2, -2, -2, -2, -2,
+ -1, -1, -1, -1, -1, -1,
+ 0, 0, 0, 0, 0, 0,
+ 1, 1, 1, 1, 1, 1,
+ 2, 2, 2, 2, 2, 2,
+ 3, 3, 3, 3, 3, 3,
+ 4, 4, 4, 4, 4, 4,
+ 5, 5, 5, 5, 5, 5,
+ 6, 6, 6, 6, 6, 6,
+ 7, 7, 7, 7, 7, 7,
+ 8, 8, 8, 8, 8, 8,
+ 9, 9, 9, 9, 9, 9,
+ ],
+ input1: [
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ -3, -2, -1, 1, 2, 3,
+ ],
+ output: [
+ 2, 3, 6, -6, -3, -2,
+ 1, 2, 5, -5, -3, -2,
+ 1, 2, 4, -4, -2, -2,
+ 1, 1, 3, -3, -2, -1,
+ 0, 1, 2, -2, -1, -1,
+ 0, 0, 1, -1, -1, -1,
+ 0, 0, 0, 0, 0, 0,
+ -1, -1, -1, 1, 0, 0,
+ -1, -1, -2, 2, 1, 0,
+ -1, -2, -3, 3, 1, 1,
+ -2, -2, -4, 4, 2, 1,
+ -2, -3, -5, 5, 2, 1,
+ -2, -3, -6, 6, 3, 2,
+ -3, -4, -7, 7, 3, 2,
+ -3, -4, -8, 8, 4, 2,
+ -3, -5, -9, 9, 4, 3,
+ ],
+})