diff options
Diffstat (limited to 'tensorflow/core/kernels/resource_variable_ops.cc')
-rw-r--r-- | tensorflow/core/kernels/resource_variable_ops.cc | 29 |
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); |