tf.summary.trace_on