Home
last modified time | relevance | path

Searched refs: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
370 struct reducer_traits<AvgPoolMeanReducer<float>, Device> { argument
385 struct reducer_traits<AvgPoolMeanReducer<float>, GpuDevice> {
402 internal::AvgPoolMeanReducer<typename internal::remove_const<
412 internal::AvgPoolMeanReducer<typename internal::remove_const<
469 internal::AvgPoolMeanReducer<CoeffReturnType> mean_with_nan;
519 internal::AvgPoolMeanReducer<float>, const Eigen::array<int, 1>,
529 internal::AvgPoolMeanReducer<float>,
597 internal::AvgPoolMeanReducer<CoeffReturnType> mean_with_nan;