From 1fe68ce0f4f7ef020cc52d1cc9963dd344fccba0 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 28 Mar 2018 10:15:46 -0700 Subject: [PATCH] internal change PiperOrigin-RevId: 190789794 --- tensorflow/core/framework/resource_mgr.h | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/tensorflow/core/framework/resource_mgr.h b/tensorflow/core/framework/resource_mgr.h index 9a45843..c84ea3b 100644 --- a/tensorflow/core/framework/resource_mgr.h +++ b/tensorflow/core/framework/resource_mgr.h @@ -319,14 +319,13 @@ class IsResourceInitialized : public OpKernel { // specified type. The type will be a part of the generated op name. // TODO(apassos): figure out how to get non-cpu-allocated tensors to work // through constant folding so this doesn't have to be marked as stateful. -#define REGISTER_RESOURCE_HANDLE_OP(Type) \ - REGISTER_OP(#Type "HandleOp") \ - .Attr("container: string = ''") \ - .Attr("shared_name: string = ''") \ - .Output("resource: resource") \ - .SetIsStateful() \ - .SetShapeFn(tensorflow::shape_inference::ScalarShape) \ - .Doc("Creates a handle to a " #Type) +#define REGISTER_RESOURCE_HANDLE_OP(Type) \ + REGISTER_OP(#Type "HandleOp") \ + .Attr("container: string = ''") \ + .Attr("shared_name: string = ''") \ + .Output("resource: resource") \ + .SetIsStateful() \ + .SetShapeFn(tensorflow::shape_inference::ScalarShape) // Utility op kernel to produce a handle to a resource of type T. template -- 2.7.4