Searched defs:AvgPoolMeanReducer (Results 1 – 1 of 1) sorted by relevance
232 struct AvgPoolMeanReducer { struct242 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE AvgPoolMeanReducer() : scalarCount_(0) { in AvgPoolMeanReducer() argument251 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reduce(const T t, T* accum) { in reduce()258 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T initialize() const { in initialize()262 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T finalize(const T accum) const { in finalize()293 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacket(const Packet& p, in reducePacket()299 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacketWithType( in reducePacketWithType()309 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet initializePacket() const { in initializePacket()336 struct reducer_traits<AvgPoolMeanReducer<float>, Device> { argument