Method gpu_boost

tensorflow/python/eager/function_test.py:2257–2258

(x)

              'api_preferred_device': 'GPU'
          })
      def gpu_boost(x):
        return math_ops.add(x, 4.0)

      x = constant_op.constant(1.0)