tests/nnapi/specs/Ex/zeros_like_ex_2D_float.mod.py

   1 # model
   2 model = Model()
   3 i1 = Input("op1", "TENSOR_FLOAT32", "{2, 3}")
   4 i2 = Output("op2", "TENSOR_FLOAT32", "{2, 3}")
   5 model = model.Operation("ZEROS_LIKE_EX", i1).To(i2)
   6
   7 # Example 1. Input in operand 0,
   8 input0 = {i1: # input 0
   9           [-2.0, -1.0, 0.0, 1.0, 2.0, 3.0]}
  10
  11 output0 = {i2: # output 0
  12            [0.0, 0.0, 0.0, 0.0, 0.0, 0.0]}
  13
  14 # Instantiate an example
  15 Example((input0, output0))