diff options
Diffstat (limited to 'nn/runtime/test/specs/V1_3/tile_quant8_signed.mod.py')
-rw-r--r-- | nn/runtime/test/specs/V1_3/tile_quant8_signed.mod.py | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/nn/runtime/test/specs/V1_3/tile_quant8_signed.mod.py b/nn/runtime/test/specs/V1_3/tile_quant8_signed.mod.py new file mode 100644 index 000000000..6349ee5e0 --- /dev/null +++ b/nn/runtime/test/specs/V1_3/tile_quant8_signed.mod.py @@ -0,0 +1,90 @@ +# +# Copyright (C) 2019 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. +# + +input0 = Input("input0", "TENSOR_FLOAT32", "{3}") +multipliers = Input("multipliers", "TENSOR_INT32", "{1}") +output0 = Output("output0", "TENSOR_FLOAT32", "{6}") + +model = Model().Operation("TILE", input0, multipliers).To(output0) + +input_values = [1.2, -3.4, 5.6] +multiplier_values = [2] +output_values = input_values + input_values + +quant8_signed = DataTypeConverter().Identify({ + input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], + output0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], +}) + +Example({ + input0: input_values, + multipliers: multiplier_values, + output0: output_values, +}).AddVariations(quant8_signed, includeDefault=False) + +####################################################### + +input0 = Input("input0", "TENSOR_FLOAT32", "{2, 3}") +multipliers = Input("multipliers", "TENSOR_INT32", "{2}") +output0 = Output("output0", "TENSOR_FLOAT32", "{4, 3}") + +model = Model().Operation("TILE", input0, multipliers).To(output0) + +input_values = [11, 12, 13, + 21, 22, 23] +multiplier_values = [2, 1] +output_values = [11, 12, 13, + 21, 22, 23, + 11, 12, 13, + 21, 22, 23] + +quant8_signed = DataTypeConverter().Identify({ + input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], + output0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], +}) + +Example({ + input0: input_values, + multipliers: multiplier_values, + output0: output_values, +}).AddVariations(quant8_signed, includeDefault=False) + +####################################################### + +input0 = Input("input0", "TENSOR_FLOAT32", "{1, 2, 3}") +multipliers = Input("multipliers", "TENSOR_INT32", "{3}") +output0 = Output("output0", "TENSOR_FLOAT32", "{2, 6, 3}") + +model = Model().Operation("TILE", input0, multipliers).To(output0) + +input_values = [11, 12, 13, + 21, 22, 23] +multiplier_values = [2, 3, 1] +output_values = [11, 12, 13, 21, 22, 23, 11, 12, 13, + 21, 22, 23, 11, 12, 13, 21, 22, 23, + 11, 12, 13, 21, 22, 23, 11, 12, 13, + 21, 22, 23, 11, 12, 13, 21, 22, 23] + +quant8_signed = DataTypeConverter().Identify({ + input0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], + output0: ["TENSOR_QUANT8_ASYMM_SIGNED", 0.5, -1], +}) + +Example({ + input0: input_values, + multipliers: multiplier_values, + output0: output_values, +}).AddVariations(quant8_signed, includeDefault=False) |