diff options
Diffstat (limited to 'tensorflow/core/kernels/sparse_conditional_accumulator_op.cc')
-rw-r--r-- | tensorflow/core/kernels/sparse_conditional_accumulator_op.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/sparse_conditional_accumulator_op.cc b/tensorflow/core/kernels/sparse_conditional_accumulator_op.cc index 80bc1f1934..1e542a26a7 100644 --- a/tensorflow/core/kernels/sparse_conditional_accumulator_op.cc +++ b/tensorflow/core/kernels/sparse_conditional_accumulator_op.cc @@ -34,8 +34,8 @@ class SparseConditionalAccumulatorOp : public ConditionalAccumulatorBaseOp { Creator GetCreator() const override { return [this](ConditionalAccumulatorBase** ret) { SparseConditionalAccumulator<Device, T>* accumulator = - new SparseConditionalAccumulator<Device, T>(dtype_, shape_, - cinfo_.name()); + new SparseConditionalAccumulator<Device, T>( + dtype_, shape_, cinfo_.name(), reduction_type_); *ret = accumulator; return Status::OK(); }; |