diff options
Diffstat (limited to 'res/TensorFlowTests/UNIT_AvgPool_001/test.pbtxt')
-rw-r--r-- | res/TensorFlowTests/UNIT_AvgPool_001/test.pbtxt | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/res/TensorFlowTests/UNIT_AvgPool_001/test.pbtxt b/res/TensorFlowTests/UNIT_AvgPool_001/test.pbtxt new file mode 100644 index 000000000..faf117ca4 --- /dev/null +++ b/res/TensorFlowTests/UNIT_AvgPool_001/test.pbtxt @@ -0,0 +1,74 @@ +node { + name: "placeholder" + op: "Placeholder" + attr { + key: "dtype" + value { + type: DT_FLOAT + } + } + attr { + key: "shape" + value { + shape { + dim { + size: 1 + } + dim { + size: 4 + } + dim { + size: 4 + } + dim { + size: 1 + } + } + } + } +} +node { + name: "avgpool2d" + op: "AvgPool" + input: "placeholder" + attr { + key: "T" + value { + type: DT_FLOAT + } + } + attr { + key: "data_format" + value { + s: "NHWC" + } + } + attr { + key: "ksize" + value { + list { + i: 1 + i: 2 + i: 2 + i: 1 + } + } + } + attr { + key: "padding" + value { + s: "VALID" + } + } + attr { + key: "strides" + value { + list { + i: 1 + i: 2 + i: 2 + i: 1 + } + } + } +} |