Searched refs:SpatialAvgPooling (Results 1 – 4 of 4) sorted by relevance
28 struct SpatialAvgPooling { struct35 To32Bit(output).swap_layout().device(d) = Eigen::SpatialAvgPooling( in operator() argument40 output.swap_layout().device(d) = Eigen::SpatialAvgPooling( in operator()
168 functor::SpatialAvgPooling<Device, T>()( in Compute()186 void SpatialAvgPooling<GPUDevice, T>::operator()( \191 extern template struct SpatialAvgPooling<GPUDevice, T>;
34 template struct functor::SpatialAvgPooling<GPUDevice, T>;
421 SpatialAvgPooling(const Input& input, DenseIndex patchRows,