diff options
| author |  Jianwei Xie <xiejw@google.com> | 2018-03-29 10:50:46 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-29 10:52:55 -0700 |
| commit | 63dffd5a3bc4e94e74cb140cbf7a68e0e5644ad6 (patch) | |
| tree | 087e334cd7e71d44cdcd1ce5ebab483768fcf474 /tensorflow/core/kernels/snapshot_op_gpu.cu.cc | |
| parent | 9fbb5b3b8fef1caa2ee2ca4a0f8dde900d1f2aa5 (diff) | |
Automated g4 rollback of changelist 190858242
PiperOrigin-RevId: 190953197
Diffstat (limited to 'tensorflow/core/kernels/snapshot_op_gpu.cu.cc')
| -rw-r--r-- | tensorflow/core/kernels/snapshot_op_gpu.cu.cc | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/tensorflow/core/kernels/snapshot_op_gpu.cu.cc b/tensorflow/core/kernels/snapshot_op_gpu.cu.cc index 52070be838..e4e3bd5220 100644 --- a/tensorflow/core/kernels/snapshot_op_gpu.cu.cc +++ b/tensorflow/core/kernels/snapshot_op_gpu.cu.cc @@ -24,13 +24,10 @@ limitations under the License. namespace tensorflow { typedef Eigen::GpuDevice GPUDevice; -#define REGISTER_KERNEL(TYPE) \ - REGISTER_KERNEL_BUILDER( \ - Name("Snapshot").Device(DEVICE_GPU).TypeConstraint<TYPE>("T"), \ - SnapshotOp<GPUDevice, TYPE>); +// Definition of the GPU implementations declared in softsign_op.cc. +#define DEFINE_GPU_KERNELS(T) template struct functor::Snapshot<GPUDevice, T>; -TF_CALL_POD_TYPES(REGISTER_KERNEL); -#undef REGISTER_KERNEL +TF_CALL_POD_TYPES(DEFINE_GPU_KERNELS); } // namespace tensorflow |