1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
# TF_SMALL_NET_0003/test.pbtxt is create with below script
# Version info
# - Tensorflow : 1.13.1
# - Python : 3.5.2
import tensorflow as tf
input = tf.placeholder(tf.float32, [1, 3, 3, 5])
filter = tf.constant(1.0, shape=[2, 2, 5, 1])
conv = tf.nn.conv2d(input, filter=filter, strides=[1, 1, 1, 1], padding='SAME')
fbn = tf.nn.fused_batch_norm(conv, scale=[1.0], offset=[0.0], mean=[0.0], variance=[1.0], is_training=False)
print(tf.get_default_graph().as_graph_def())
|