diff options
Diffstat (limited to 'tensorflow/core/kernels/resource_variable_ops.h')
-rw-r--r-- | tensorflow/core/kernels/resource_variable_ops.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/resource_variable_ops.h b/tensorflow/core/kernels/resource_variable_ops.h index 8cae5d21f0..9b60106f13 100644 --- a/tensorflow/core/kernels/resource_variable_ops.h +++ b/tensorflow/core/kernels/resource_variable_ops.h @@ -28,6 +28,15 @@ class ReadVariableOp : public OpKernel { DataType dtype_; }; +class DestroyResourceOp : public OpKernel { + public: + explicit DestroyResourceOp(OpKernelConstruction* ctx); + void Compute(OpKernelContext* ctx) override; + + private: + bool ignore_lookup_error_; +}; + } // namespace tensorflow #endif // TENSORFLOW_CORE_KERNELS_RESOURCE_VARIABLE_OPS_H_ |