18c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen#
28c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# Copyright (C) 2017 The Android Open Source Project
38c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen#
48c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# Licensed under the Apache License, Version 2.0 (the "License");
58c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# you may not use this file except in compliance with the License.
68c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# You may obtain a copy of the License at
78c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen#
88c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen#      http://www.apache.org/licenses/LICENSE-2.0
98c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen#
108c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# Unless required by applicable law or agreed to in writing, software
118c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# distributed under the License is distributed on an "AS IS" BASIS,
128c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
138c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# See the License for the specific language governing permissions and
148c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# limitations under the License.
158c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen#
168c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen
178c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# model
188c0aac93226fb68c12fb01185f9bb37640b8455eDong Chenmodel = Model()
198c0aac93226fb68c12fb01185f9bb37640b8455eDong Cheni1 = Input("op1", "TENSOR_QUANT8_ASYMM", "{1, 2, 2, 1}, 0.5f, 0") # input 0
208c0aac93226fb68c12fb01185f9bb37640b8455eDong Cheni2 = Output("op2", "TENSOR_QUANT8_ASYMM", "{1, 2, 2, 1}, 0.5f, 0") # output 0
218c0aac93226fb68c12fb01185f9bb37640b8455eDong Chenmodel = model.Operation("RELU6", i1).To(i2)
228c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen
238c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# Example 1. Input in operand 0,
248c0aac93226fb68c12fb01185f9bb37640b8455eDong Cheninput0 = {i1: # input 0
258c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen          [0, 1, 11, 12]}
268c0aac93226fb68c12fb01185f9bb37640b8455eDong Chenoutput0 = {i2: # output 0
278c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen          [0, 1, 11, 12]}
288c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# Instantiate an example
298c0aac93226fb68c12fb01185f9bb37640b8455eDong ChenExample((input0, output0))
308c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen
318c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# Example 2. Input in operand 0,
328c0aac93226fb68c12fb01185f9bb37640b8455eDong Cheninput1 = {i1: # input 0
338c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen          [13, 14, 254, 255]}
348c0aac93226fb68c12fb01185f9bb37640b8455eDong Chenoutput1 = {i2: # output 0
358c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen          [12, 12, 12, 12]}
368c0aac93226fb68c12fb01185f9bb37640b8455eDong Chen# Instantiate an example
378c0aac93226fb68c12fb01185f9bb37640b8455eDong ChenExample((input1, output1))
38