res/TensorFlowTests/UNIT_CustomOp_000/customop.conf


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

# defining a list of custom ops for this compilation
custom_op {
  name: "my/customOp/000"
  op: "new_custom_op"
  attr {
    key: "dtype"
    value {
      type: DT_FLOAT
    }
  }
  attr {
    key: "output_shape"
    value {
      shape {
        dim { size: 1 }
        dim { size: 2 }
        dim { size: 1 }
        dim { size: 2 }
      }
    }
  }
}