diff options
Diffstat (limited to 'tensorflow/core/ops/data_flow_ops.cc')
| -rw-r--r-- | tensorflow/core/ops/data_flow_ops.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/ops/data_flow_ops.cc b/tensorflow/core/ops/data_flow_ops.cc index eed0bce174..ffab8ad661 100644 --- a/tensorflow/core/ops/data_flow_ops.cc +++ b/tensorflow/core/ops/data_flow_ops.cc @@ -419,6 +419,7 @@ REGISTER_OP("ConditionalAccumulator") .Attr("shape: shape") .Attr("container: string = ''") .Attr("shared_name: string = ''") + .Attr("reduction_type: { 'MEAN', 'SUM' } = 'MEAN' ") .SetIsStateful() .SetShapeFn([](InferenceContext* c) { c->set_output(0, c->Vector(2)); @@ -456,6 +457,7 @@ REGISTER_OP("SparseConditionalAccumulator") .Attr("shape: shape") .Attr("container: string = ''") .Attr("shared_name: string = ''") + .Attr("reduction_type: { 'MEAN', 'SUM' } = 'MEAN' ") .SetIsStateful() .SetShapeFn([](InferenceContext* c) { c->set_output(0, c->Vector(2)); |