aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/resource_variable_ops.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/resource_variable_ops.cc')
-rw-r--r--tensorflow/core/kernels/resource_variable_ops.cc29
1 files changed, 12 insertions, 17 deletions
diff --git a/tensorflow/core/kernels/resource_variable_ops.cc b/tensorflow/core/kernels/resource_variable_ops.cc
index af921e4815..c5292e1ae1 100644
--- a/tensorflow/core/kernels/resource_variable_ops.cc
+++ b/tensorflow/core/kernels/resource_variable_ops.cc
@@ -174,25 +174,20 @@ REGISTER_KERNEL_BUILDER(Name("VariableShape")
#endif // GOOGLE_CUDA
-class DestroyResourceOp : public OpKernel {
- public:
- explicit DestroyResourceOp(OpKernelConstruction* ctx) : OpKernel(ctx) {
- OP_REQUIRES_OK(ctx,
- ctx->GetAttr("ignore_lookup_error", &ignore_lookup_error_));
- }
+DestroyResourceOp::DestroyResourceOp(OpKernelConstruction* ctx)
+ : OpKernel(ctx) {
+ OP_REQUIRES_OK(ctx,
+ ctx->GetAttr("ignore_lookup_error", &ignore_lookup_error_));
+}
- void Compute(OpKernelContext* ctx) override {
- const ResourceHandle& p = HandleFromInput(ctx, 0);
- Status status = DeleteResource(ctx, p);
- if (ignore_lookup_error_ && errors::IsNotFound(status)) {
- return;
- }
- OP_REQUIRES_OK(ctx, status);
+void DestroyResourceOp::Compute(OpKernelContext* ctx) {
+ const ResourceHandle& p = HandleFromInput(ctx, 0);
+ Status status = DeleteResource(ctx, p);
+ if (ignore_lookup_error_ && errors::IsNotFound(status)) {
+ return;
}
-
- private:
- bool ignore_lookup_error_;
-};
+ OP_REQUIRES_OK(ctx, status);
+}
REGISTER_KERNEL_BUILDER(Name("DestroyResourceOp").Device(DEVICE_CPU),
DestroyResourceOp);