import tensorflow as tf

                        X = tf.identity(X)
                else:
                    # For prediction, do batch norm with computed moving mean & variance from training
                    # Don't update moving averages if predicting
                    (X, _, _) =  tf.nn.fused_batch_norm(X, scale, offset, mean=moving_mean, variance=moving_variance,
                                                            epsilon=epsilon, is_training=False)
            else:
                (X, _, _) =  tf.nn.fused_batch_norm(X, scale, offset, epsilon=epsilon, is_training=True)
