res/TensorFlowPythonModels/examples/tconv-bn/__init__.py

   1 import tensorflow as tf
   2 import numpy as np
   3
   4 input_ = tf.compat.v1.placeholder(tf.float32, shape=(1, 2, 2, 1), name="Hole")
   5 W = np.ones(9).reshape((3, 3, 1, 1))
   6 filter_ = tf.compat.v1.constant(W, dtype=tf.float32)
   7 tconv_ = tf.compat.v1.nn.conv2d_transpose(
   8     input_, filter_, output_shape=(1, 4, 4, 1), strides=[1, 1, 1, 1], padding='VALID')
   9
  10 scale_ = tf.compat.v1.constant([1.0177339315414429], dtype=tf.float32)
  11 offset_ = tf.compat.v1.constant([0.015628524124622345], dtype=tf.float32)
  12 mean_ = tf.compat.v1.constant([1.027155211195349693], dtype=tf.float32)
  13 variance_ = tf.compat.v1.constant([0.25580066442489624], dtype=tf.float32)
  14 bn_out, _, _ = tf.compat.v1.nn.fused_batch_norm(
  15     tconv_,
  16     scale_,
  17     offset_,
  18     mean=mean_,
  19     variance=variance_,
  20     epsilon=0.0010000000474974513,
  21     is_training=False)
  22 '''
  23 python ../../compiler/tf2tfliteV2/tf2tfliteV2.py --v1 \
  24 -i tconv-bn.pbtxt \
  25 -o tconv-bn.tflite \
  26 -I Hole -O FusedBatchNorm
  27 '''