summaryrefslogtreecommitdiff
path: root/res/TensorFlowTests/NET_0004/test.py
diff options
context:
space:
mode:
Diffstat (limited to 'res/TensorFlowTests/NET_0004/test.py')
-rw-r--r--res/TensorFlowTests/NET_0004/test.py16
1 files changed, 16 insertions, 0 deletions
diff --git a/res/TensorFlowTests/NET_0004/test.py b/res/TensorFlowTests/NET_0004/test.py
new file mode 100644
index 000000000..a0c790d79
--- /dev/null
+++ b/res/TensorFlowTests/NET_0004/test.py
@@ -0,0 +1,16 @@
+# TF_SMALL_NET_0004/test.pbtxt is create with below script
+
+# Version info
+# - Tensorflow : 1.13.1
+# - Python : 3.5.2
+
+import tensorflow as tf
+
+input0 = tf.placeholder(tf.float32, [1, 3, 3, 5])
+filter0 = tf.constant(1.0, shape=[2, 2, 5, 2])
+dconv = tf.nn.depthwise_conv2d(input0, filter0, [1, 1, 1, 1], 'SAME')
+const = tf.constant(2.0, shape=[10])
+fbn = tf.nn.fused_batch_norm(
+ x=dconv, scale=const, offset=const, mean=const, variance=const, is_training=False)
+
+print(tf.get_default_graph().as_graph_def())