tf.debugging.enable_check_numerics