diff options
Diffstat (limited to 'res/TensorFlowTests/UNIT_Conv2D_001/test.pbtxt')
-rw-r--r-- | res/TensorFlowTests/UNIT_Conv2D_001/test.pbtxt | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/res/TensorFlowTests/UNIT_Conv2D_001/test.pbtxt b/res/TensorFlowTests/UNIT_Conv2D_001/test.pbtxt new file mode 100644 index 000000000..6eb539be8 --- /dev/null +++ b/res/TensorFlowTests/UNIT_Conv2D_001/test.pbtxt @@ -0,0 +1,72 @@ +node { + name: "ifm" + op: "Placeholder" + attr { + key: "dtype" + value { type: DT_FLOAT } + } + attr { + key: "shape" + value { + shape { + dim { size: 1 } + dim { size: 5 } + dim { size: 5 } + dim { size: 3 } + } + } + } +} +node { + name: "ker" + op: "Const" + attr { + key: "dtype" + value { type: DT_FLOAT } + } + attr { + key: "value" + value { + tensor { + dtype: DT_FLOAT + tensor_shape { + dim { size: 2 } + dim { size: 2 } + dim { size: 3 } + dim { size: 2 } + } + float_val: 1.1 + } + } + } +} +node { + name: "ofm" + op: "Conv2D" + input: "ifm" + input: "ker" + attr { + key: "T" + value { type: DT_FLOAT } + } + attr { + key: "data_format" + value { s: "NHWC" } + } + attr { + key: "dilations" + value { + list { i: 1 i: 1 i: 1 i: 1 } + } + } + attr { + key: "padding" + value { s: "SAME" } + } + attr { + key: "strides" + value { + list { i: 1 i: 2 i: 2 i: 1 } + } + } +} |