Backward computation for max pooling. More...
#include <include/dlprim/core/pool.hpp>
Inheritance diagram for dlprim::core::MaxPooling2DBackward:
Static Public Member Functions | |
| static std::unique_ptr< MaxPooling2DBackward > | create (Context &ctx, int kernel[2], int pad[2], int stride[2], DataType dt=float_data) |
| Create pooling backward computation, See Pooling2DForward for details. | |
| static std::unique_ptr< MaxPooling2DBackward > | create_global (Context &ctx, Shape const &in_shape, DataType dt=float_data) |
| Create global pooling. | |
Additional Inherited Members | |
 Public Member Functions inherited from dlprim::core::Pooling2DBackwardBase | |
| virtual size_t | workspace ()=0 |
| get workspace | |
| virtual void | enqueue (Tensor &X, Tensor &dX, Tensor &dY, float factor, ExecutionContext const &e)=0 |
| X for max pooling to detect the neuron giving max signal. More... | |
Detailed Description
Backward computation for max pooling.
The documentation for this class was generated from the following file:
- include/dlprim/core/pool.hpp
