Home | History | Annotate | Download | only in V1_0
      1 #
      2 # Copyright (C) 2017 The Android Open Source Project
      3 #
      4 # Licensed under the Apache License, Version 2.0 (the "License");
      5 # you may not use this file except in compliance with the License.
      6 # You may obtain a copy of the License at
      7 #
      8 #      http://www.apache.org/licenses/LICENSE-2.0
      9 #
     10 # Unless required by applicable law or agreed to in writing, software
     11 # distributed under the License is distributed on an "AS IS" BASIS,
     12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13 # See the License for the specific language governing permissions and
     14 # limitations under the License.
     15 #
     16 
     17 # model
     18 model = Model()
     19 i1 = Input("op1", "TENSOR_QUANT8_ASYMM", "{1, 2, 2, 1}, 0.5f, 0") # input 0
     20 i2 = Output("op2", "TENSOR_QUANT8_ASYMM", "{1, 2, 2, 1}, 0.5f, 0") # output 0
     21 model = model.Operation("RELU6", i1).To(i2)
     22 
     23 # Example 1. Input in operand 0,
     24 input0 = {i1: # input 0
     25           [0, 1, 11, 12]}
     26 output0 = {i2: # output 0
     27           [0, 1, 11, 12]}
     28 # Instantiate an example
     29 Example((input0, output0))
     30 
     31 # Example 2. Input in operand 0,
     32 input1 = {i1: # input 0
     33           [13, 14, 254, 255]}
     34 output1 = {i2: # output 0
     35           [12, 12, 12, 12]}
     36 # Instantiate an example
     37 Example((input1, output1))
     38