Searched defs:AvgPoolMeanReducer (Results 1 – 1 of 1) sorted by relevance
276 struct AvgPoolMeanReducer { struct286 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE AvgPoolMeanReducer() : scalarCount_(0) { in AvgPoolMeanReducer() argument291 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reduce(const T t, T* accum) { in reduce()298 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T initialize() const { in initialize()302 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T finalize(const T accum) const { in finalize()333 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacket(const Packet& p, in reducePacket()339 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacketWithType( in reducePacketWithType()372 struct reducer_traits<AvgPoolMeanReducer<float>, Device> { argument