summaryrefslogtreecommitdiff
path: root/nn/runtime/test/specs
diff options
context:
space:
mode:
Diffstat (limited to 'nn/runtime/test/specs')
-rw-r--r--nn/runtime/test/specs/V1_1/mean_b155508675.mod.py29
-rw-r--r--nn/runtime/test/specs/V1_1/squeeze_b155238914.mod.py (renamed from nn/runtime/test/specs/V1_2/squeeze_b155238914.mod.py)17
-rw-r--r--nn/runtime/test/specs/V1_1/strided_slice_b155662254.mod.py29
-rw-r--r--nn/runtime/test/specs/V1_2/argmax_b155660285.mod.py29
-rw-r--r--nn/runtime/test/specs/V1_2/argmin_b155660285.mod.py29
5 files changed, 123 insertions, 10 deletions
diff --git a/nn/runtime/test/specs/V1_1/mean_b155508675.mod.py b/nn/runtime/test/specs/V1_1/mean_b155508675.mod.py
new file mode 100644
index 000000000..99bfbe63d
--- /dev/null
+++ b/nn/runtime/test/specs/V1_1/mean_b155508675.mod.py
@@ -0,0 +1,29 @@
+#
+# 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 operands
+op1 = Input("op1", ["TENSOR_FLOAT32", [2, 2, 1]])
+op2 = Output("op2", ["TENSOR_FLOAT32", [1]])
+
+# Model operations
+model = Model()
+model.Operation("MEAN", op1, [0, 1, 2], 0).To(op2)
+
+# Example
+Example({
+ op1: [2.0, 3.0, 3.0, 4.0],
+ op2: [3.0],
+}, model=model)
diff --git a/nn/runtime/test/specs/V1_2/squeeze_b155238914.mod.py b/nn/runtime/test/specs/V1_1/squeeze_b155238914.mod.py
index 70e41c6d3..75ac4381f 100644
--- a/nn/runtime/test/specs/V1_2/squeeze_b155238914.mod.py
+++ b/nn/runtime/test/specs/V1_1/squeeze_b155238914.mod.py
@@ -14,19 +14,16 @@
# limitations under the License.
#
# Model operands
-op0 = Input("op0", ["TENSOR_FLOAT32", [9, 1, 1]])
-op1 = Output("op1", ["TENSOR_FLOAT32", [9, 1, 1]])
-op5 = Parameter("op5", ["TENSOR_FLOAT32", [1, 1, 1]], [0])
-op7 = Parameter("op7", ["TENSOR_INT32", [0]], value=None) # omitted
-op8 = Internal("op8", ["TENSOR_FLOAT32", []])
+op1 = Input("op1", ["TENSOR_FLOAT32", [1, 1, 1]], [0])
+op2 = Parameter("op2", ["TENSOR_INT32", [0]], value=None) # omitted
+op3 = Output("op3", ["TENSOR_FLOAT32", [1]])
# Model operations
model = Model()
-model.Operation("FLOOR", op0).To(op1)
-model.Operation("SQUEEZE", op5, op7).To(op8)
+model.Operation("SQUEEZE", op1, op2).To(op3)
# Example
Example({
- op0: [0, 0, 0, 0, 0, 0, 0, 0, 0],
- op1: [0, 0, 0, 0, 0, 0, 0, 0, 0],
-}, model=model).DisableLifeTimeVariation()
+ op1: [0],
+ op3: [0]
+}, model=model)
diff --git a/nn/runtime/test/specs/V1_1/strided_slice_b155662254.mod.py b/nn/runtime/test/specs/V1_1/strided_slice_b155662254.mod.py
new file mode 100644
index 000000000..908f8563d
--- /dev/null
+++ b/nn/runtime/test/specs/V1_1/strided_slice_b155662254.mod.py
@@ -0,0 +1,29 @@
+#
+# 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 operands
+op1 = Input("op1", ["TENSOR_FLOAT32", [4]])
+op2 = Output("op2", ["TENSOR_FLOAT32", [1]])
+
+# Model operations
+model = Model()
+model.Operation("STRIDED_SLICE", op1, [0], [1], [1], 0, 0, 1).To(op2)
+
+# Example
+Example({
+ op1: [0.0, 1.0, 2.0, 3.0],
+ op2: [0.0],
+}, model=model)
diff --git a/nn/runtime/test/specs/V1_2/argmax_b155660285.mod.py b/nn/runtime/test/specs/V1_2/argmax_b155660285.mod.py
new file mode 100644
index 000000000..6247e1519
--- /dev/null
+++ b/nn/runtime/test/specs/V1_2/argmax_b155660285.mod.py
@@ -0,0 +1,29 @@
+#
+# 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 operands
+op1 = Input("op1", ["TENSOR_FLOAT32", [3]])
+op2 = Output("op2", ["TENSOR_INT32", [1]])
+
+# Model operations
+model = Model()
+model.Operation("ARGMAX", op1, 0).To(op2)
+
+# Example
+Example({
+ op1: [1.0, 2.0, 4.0],
+ op2: [2],
+}, model=model)
diff --git a/nn/runtime/test/specs/V1_2/argmin_b155660285.mod.py b/nn/runtime/test/specs/V1_2/argmin_b155660285.mod.py
new file mode 100644
index 000000000..a59500a6a
--- /dev/null
+++ b/nn/runtime/test/specs/V1_2/argmin_b155660285.mod.py
@@ -0,0 +1,29 @@
+#
+# 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 operands
+op1 = Input("op1", ["TENSOR_FLOAT32", [3]])
+op2 = Output("op2", ["TENSOR_INT32", [1]])
+
+# Model operations
+model = Model()
+model.Operation("ARGMIN", op1, 0).To(op2)
+
+# Example
+Example({
+ op1: [4.0, 2.0, 1.0],
+ op2: [2],
+}, model=model)