Home
last modified time | relevance | path

Searched defs:AvgPoolMeanReducer (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h276 struct AvgPoolMeanReducer { struct
285 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE AvgPoolMeanReducer() : scalarCount_(0) { in AvgPoolMeanReducer() argument
290 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reduce(const T t, T* accum) { in reduce()
297 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T initialize() const { in initialize()
301 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE T finalize(const T accum) const { in finalize()
331 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacket(const Packet& p, in reducePacket()
337 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacketWithType( in reducePacketWithType()
370 struct reducer_traits<AvgPoolMeanReducer<float>, Device> { argument