aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/resource_variable_ops.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/resource_variable_ops.h')
-rw-r--r--tensorflow/core/kernels/resource_variable_ops.h9
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_