Home | History | Annotate | Download | only in specs
      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 = Model()
     18 i1 = Input("op1", "TENSOR_FLOAT32", "{1, 2, 2, 2}") # depth_in = 2
     19 f1 = Parameter("op2", "TENSOR_FLOAT32", "{1, 2, 2, 2}", [.25, 0, .25, 1, .25, 0, .25, 1]) # depth_out = 2
     20 b1 = Parameter("op3", "TENSOR_FLOAT32", "{2}", [100, 200]) # depth_out = 2
     21 pad0 = Int32Scalar("pad0", 0)
     22 act = Int32Scalar("act", 0)
     23 stride = Int32Scalar("stride", 1)
     24 cm = Int32Scalar("channelMultiplier", 1)
     25 output = Output("op4", "TENSOR_FLOAT32", "{1, 1, 1, 2}")
     26 
     27 model = model.Operation("DEPTHWISE_CONV_2D",
     28                         i1, f1, b1,
     29                         pad0, pad0, pad0, pad0,
     30                         stride, stride,
     31                         cm, act).To(output)
     32 
     33 # Example 1. Input in operand 0,
     34 input0 = {
     35     i1: [ # input 0
     36      10, 21, 10, 22,
     37      10, 23, 10, 24],
     38   }
     39 # (i1 (conv) f1) + b1
     40 output0 = {output: # output 0
     41            [110, 246]}
     42 
     43 # Instantiate an example
     44 Example((input0, output0))
     45