diff options
Diffstat (limited to 'tensorflow/core/kernels/sparse_conditional_accumulator.h')
-rw-r--r-- | tensorflow/core/kernels/sparse_conditional_accumulator.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/sparse_conditional_accumulator.h b/tensorflow/core/kernels/sparse_conditional_accumulator.h index 11149c4d16..a4453bd7ab 100644 --- a/tensorflow/core/kernels/sparse_conditional_accumulator.h +++ b/tensorflow/core/kernels/sparse_conditional_accumulator.h @@ -50,10 +50,10 @@ class SparseConditionalAccumulator public: SparseConditionalAccumulator(const DataType& dtype, const PartialTensorShape& shape, - const string& name) + const string& name, const string& reduction_type) : TypedConditionalAccumulatorBase< std::tuple<const Tensor*, const Tensor*, const Tensor*>>( - dtype, shape, name) { + dtype, shape, name, reduction_type) { accum_idx_vec_ = nullptr; count_element_ = nullptr; accum_val_ = nullptr; |