TFE: Register a GPU kernel for tfe.py_func.
authorAkshay Agrawal <akshayka@google.com>
Tue, 30 Jan 2018 02:30:59 +0000 (18:30 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 30 Jan 2018 02:34:39 +0000 (18:34 -0800)
commit87a3c967973641d3b0d2a16d17add184ed967392
tree4cd133e7acd12a9b173ce172c516fcff912ee98e
parente3d99c92975efc2010d0e1e2dd4c3eb787a8d67c
TFE: Register a GPU kernel for tfe.py_func.

PiperOrigin-RevId: 183765122
tensorflow/python/BUILD
tensorflow/python/kernel_tests/py_func_test.py
tensorflow/python/lib/core/py_func.cc
tensorflow/python/ops/script_ops.py