diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-28 10:15:46 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-28 10:20:39 -0700 |
| commit | 1fe68ce0f4f7ef020cc52d1cc9963dd344fccba0 (patch) | |
| tree | 0d03e00278f1a43f6920a53d5c1229f79d53af80 | |
| parent | 2d2d4529fe3b99f25e531ee2fc0e9211341c160f (diff) | |
internal change
PiperOrigin-RevId: 190789794
| -rw-r--r-- | tensorflow/core/framework/resource_mgr.h | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/tensorflow/core/framework/resource_mgr.h b/tensorflow/core/framework/resource_mgr.h index 9a458431e7..c84ea3b034 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 <typename T> |