function wrapper to measure execution time in tf graph

tf_measure.py

def measure_time_tf(fun, name):
    def time_measuring(*args):
        start = tf.timestamp()
        result = fun(*args)
        end = tf.timestamp()
        diff_times100 = 100.0 * (end - start)
        tf.print(name, tf.cast(tf.cast(diff_times100, tf.int32), tf.float32) / 100.0)
        return result
    return time_measuring