tests/nnapi/specs/V1_2/equal_dynamic_float_nnfw.mod.py

   1 #
   2 # Copyright (C) 2018 The Android Open Source Project
   3 # Copyright (C) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
   4 #
   5 # Licensed under the Apache License, Version 2.0 (the "License");
   6 # you may not use this file except in compliance with the License.
   7 # You may obtain a copy of the License at
   8 #
   9 #      http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing, software
  12 # distributed under the License is distributed on an "AS IS" BASIS,
  13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 # See the License for the specific language governing permissions and
  15 # limitations under the License.
  16 #
  17
  18 import dynamic_tensor
  19
  20 model = Model()
  21
  22 input0_shape = [3, 4]
  23
  24 dynamic_layer = dynamic_tensor.DynamicInputGenerator(model, input0_shape, "TENSOR_FLOAT32")
  25
  26 input0 = dynamic_layer.getTestNodeInput() # first input of equal is dynamic tensor
  27
  28 # model definition
  29 input1 = Input("input1", "TENSOR_FLOAT32", "{1, 4}") # second input of Equal.
  30 output = Output("output", "TENSOR_BOOL8", "{3, 4}")
  31
  32 model = model.Operation("EQUAL", input0, input1).To(output)
  33
  34 model_input0_data = [1.1, 2.0, 2.0, 2.0, 1.1, 2.2, 2.2, 2.2, 1.0, 2.0, 2.0, 2.1]
  35 model_input1_data = [1.1 ,2.0 ,2.0 ,2.0]
  36
  37 input_data = {
  38     dynamic_layer.getModelInput() : model_input0_data,
  39     dynamic_layer.getShapeInput() : input0_shape,
  40
  41     input1 : model_input1_data
  42 }
  43
  44 output_data = {
  45     output : [True, True, True, True, True, False, False, False, False, True, True, False]
  46 }
  47
  48 # Instantiate an example
  49 Example((input_data, output_data))