diff options
| author |  Allen Lavoie <allenl@google.com> | 2017-10-16 09:53:11 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-16 09:57:40 -0700 |
| commit | 07c6faf039ff1a49a540f8caafd7a30692fcf14d (patch) | |
| tree | b2bb4fc111502b0568bfec164927e33cce9fd6e2 | |
| parent | 0a572887ffa9879d6a303aafd6de9e288776fc8f (diff) | |
Adds a host-memory GPU kernel for DestroyResourceOp.
PiperOrigin-RevId: 172337312
| -rw-r--r-- | tensorflow/core/kernels/resource_variable_ops.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/resource_variable_ops.cc b/tensorflow/core/kernels/resource_variable_ops.cc index 3cca493972..90db0c2b7b 100644 --- a/tensorflow/core/kernels/resource_variable_ops.cc +++ b/tensorflow/core/kernels/resource_variable_ops.cc @@ -200,6 +200,9 @@ class DestroyResourceOp : public OpKernel { REGISTER_KERNEL_BUILDER(Name("DestroyResourceOp").Device(DEVICE_CPU), DestroyResourceOp); +REGISTER_KERNEL_BUILDER( + Name("DestroyResourceOp").Device(DEVICE_GPU).HostMemory("resource"), + DestroyResourceOp); template <typename Device, typename T> class AssignVariableOp : public OpKernel { |