Home
last modified time | relevance | path

Searched refs:scalarCount_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h285 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE AvgPoolMeanReducer() : scalarCount_(0) { in AvgPoolMeanReducer()
293 scalarCount_++; in reduce()
302 eigen_assert(scalarCount_ > 0); in finalize()
303 return accum / T(scalarCount_); in finalize()
359 return (saccum + predux(vaccum)) / (scalarCount_ + predux(packetCount_)); in finalizeBoth()
365 int scalarCount_; member
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorFunctors.h147 MeanReducer() : scalarCount_(0), packetCount_(0) { }
152 scalarCount_++;
169 return accum / scalarCount_;
178 …return sum_op(saccum, predux(vaccum)) / (scalarCount_ + packetCount_ * unpacket_traits<Packet>::si…
182 DenseIndex scalarCount_;