diff options
Diffstat (limited to 'res/TensorFlowTests/UNIT_DepthwiseConv2dNative_000/test.pbtxt')
-rw-r--r-- | res/TensorFlowTests/UNIT_DepthwiseConv2dNative_000/test.pbtxt | 115 |
1 files changed, 115 insertions, 0 deletions
diff --git a/res/TensorFlowTests/UNIT_DepthwiseConv2dNative_000/test.pbtxt b/res/TensorFlowTests/UNIT_DepthwiseConv2dNative_000/test.pbtxt new file mode 100644 index 000000000..1d8fcaa00 --- /dev/null +++ b/res/TensorFlowTests/UNIT_DepthwiseConv2dNative_000/test.pbtxt @@ -0,0 +1,115 @@ +# A simple network that has DepthwiseConv2dNative with 'SAME' padding and strides. +# HOW TO GENERATE: +# import tensorflow as tf +# input = tf.placeholder(tf.float32, shape=[1,11,7,3], name="input") +# filter = tf.constant(1.0, shape=[2,3,3,4], dtype=tf.float32) +# dwconv = tf.nn.depthwise_conv2d_native(input,filter,[1,2,2,1],'SAME') +# tf.get_default_graph().as_graph_def() +node { + name: "input" + op: "Placeholder" + attr { + key: "dtype" + value { + type: DT_FLOAT + } + } + attr { + key: "shape" + value { + shape { + dim { + size: 1 + } + dim { + size: 11 + } + dim { + size: 7 + } + dim { + size: 3 + } + } + } + } +} +node { + name: "Const" + op: "Const" + attr { + key: "dtype" + value { + type: DT_FLOAT + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_FLOAT + tensor_shape { + dim { + size: 2 + } + dim { + size: 3 + } + dim { + size: 3 + } + dim { + size: 4 + } + } + float_val: 1.0 + } + } + } +} +node { + name: "DepthwiseConv2dNative" + op: "DepthwiseConv2dNative" + input: "input" + input: "Const" + 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 + } + } + } +} |