diff options
author | 2018-01-12 22:22:35 -0800 | |
---|---|---|
committer | 2018-01-12 22:26:32 -0800 | |
commit | 505258f652f55e3a03513810d5768faed282e45e (patch) | |
tree | 8b39acd88bebfaf079cb25c96b81ef371eed7e24 /tensorflow/core/ops/resource_variable_ops.cc | |
parent | 7022e6b62908f688cae446f1abbc667c5273db04 (diff) |
[TF] Additional CriticalSection python wrappers.
PiperOrigin-RevId: 181827605
Diffstat (limited to 'tensorflow/core/ops/resource_variable_ops.cc')
-rw-r--r-- | tensorflow/core/ops/resource_variable_ops.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/ops/resource_variable_ops.cc b/tensorflow/core/ops/resource_variable_ops.cc index a5c72f0193..f6cfbf873a 100644 --- a/tensorflow/core/ops/resource_variable_ops.cc +++ b/tensorflow/core/ops/resource_variable_ops.cc @@ -227,8 +227,8 @@ REGISTER_OP("ExecuteInCriticalSection") .Output("outputs: output_types") .Attr("f: func") .Attr("Targuments: list(type) >= 0") - .Attr("output_types: list(type) >= 1") - .Attr("output_shapes: list(shape) >= 1") + .Attr("output_types: list(type) >= 0") + .Attr("output_shapes: list(shape) >= 0") .SetShapeFn([](InferenceContext* c) { std::vector<PartialTensorShape> output_shapes; TF_RETURN_IF_ERROR(c->GetAttr("output_shapes", &output_shapes)); |