diff options
| author | Slava Shklyaev <slavash@google.com> | 2020-01-09 14:58:29 +0000 |
|---|---|---|
| committer | Slava Shklyaev <slavash@google.com> | 2020-01-13 15:43:41 +0000 |
| commit | c01d2a24ac6de9e9db8a503bb54d37abc5feafd7 (patch) | |
| tree | 4110b069b25db2d63237f2928fe363d0489fc341 /nn/runtime/test/specs/V1_3 | |
| parent | c0f6ec16fc73c7881ff0a03bc93b0eb9877be4ef (diff) | |
| download | ml-c01d2a24ac6de9e9db8a503bb54d37abc5feafd7.tar.gz | |
Add TENSOR_INT32 support for MUL
Bug: 142318767
Bug: 136735929
Test: NNT_static
Change-Id: I42d1d5874be6bc7ff11ecf33fd15ce9db539d1a3
Diffstat (limited to 'nn/runtime/test/specs/V1_3')
| -rw-r--r-- | nn/runtime/test/specs/V1_3/mul_int32.mod.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/nn/runtime/test/specs/V1_3/mul_int32.mod.py b/nn/runtime/test/specs/V1_3/mul_int32.mod.py new file mode 100644 index 000000000..d50bb40e3 --- /dev/null +++ b/nn/runtime/test/specs/V1_3/mul_int32.mod.py @@ -0,0 +1,23 @@ +# +# 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", "{1, 2, 2}") +input1 = Input("input1", "TENSOR_INT32", "{1, 2, 2}") +output = Output("output", "TENSOR_INT32", "{1, 2, 2}") +model = model.Operation("MUL", input0, input1, 0).To(output) +Example({input0: [2, -4, 8, -16], + input1: [32, -16, -8, 4], + output: [64, 64, -64, -64]}) |