()
Processes TensorFlow-relevant flags.
Processes TensorFlow-relevant flags.
Processes TensorFlow - relevant flags .
def _tf_setup_from_flags():
  """Processes TensorFlow-relevant flags."""
  if FLAGS.enable_eager_execution:
    tf.compat.v1.enable_eager_execution()
  if FLAGS.tf_xla:
    tf.config.optimizer.set_jit(True)
    fastmath.tf.set_tf_xla_forced_compile(FLAGS.tf_xla_forced_compile)
  tf.config.optimizer.set_experimental_options({
      'pin_to_host_optimization': FLAGS.tf_opt_pin_to_host,
      'layout_optimizer': FLAGS.tf_opt_layout,
  })
  tf_np.set_allow_float64(FLAGS.tf_allow_float64)
def _tf_setup_from_flags ( ) : if FLAGS . enable_eager_execution : tf . compat . v1 . enable_eager_execution ( ) if FLAGS . tf_xla : tf . config . optimizer . set_jit ( True ) fastmath . tf . set_tf_xla_forced_compile ( FLAGS . tf_xla_forced_compile ) tf . config . optimizer . set_experimental_options ( { 'pin_to_host_optimization' : FLAGS . tf_opt_pin_to_host , 'layout_optimizer' : FLAGS . tf_opt_layout , } ) tf_np . set_allow_float64 ( FLAGS . tf_allow_float64 )