Better performance with tf.function

Open in new window