mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 14:24:22 +00:00
60 lines
1.3 KiB
C++
60 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#include "layer.hpp"
|
|
|
|
namespace CUDANet::Layers {
|
|
|
|
class MaxPool2d : public CUDANet::Layer {
|
|
public:
|
|
MaxPool2d(
|
|
CUDANet::Shape input_shape,
|
|
CUDANet::Shape pool_shape,
|
|
CUDANet::Shape stride_shape,
|
|
CUDANet::Shape padding_shape,
|
|
CUDANet::Backend* backend
|
|
);
|
|
MaxPool2d(
|
|
CUDANet::Shape input_shape,
|
|
CUDANet::Shape pool_shape,
|
|
CUDANet::Shape stride_shape,
|
|
CUDANet::Shape padding_shape,
|
|
CUDANet::DType dtype,
|
|
CUDANet::Backend* backend
|
|
);
|
|
~MaxPool2d();
|
|
|
|
CUDANet::Tensor& forward(CUDANet::Tensor &input) override;
|
|
|
|
CUDANet::Shape input_shape() override;
|
|
|
|
CUDANet::Shape output_shape() override;
|
|
|
|
size_t input_size() override;
|
|
|
|
size_t output_size() override;
|
|
|
|
void set_weights(void *input) override;
|
|
|
|
size_t get_weights_size() override;
|
|
|
|
void set_biases(void *input) override;
|
|
|
|
size_t get_biases_size() override;
|
|
|
|
|
|
|
|
private:
|
|
CUDANet::Shape in_shape;
|
|
|
|
CUDANet::Shape pool_shape;
|
|
CUDANet::Shape stride_shape;
|
|
CUDANet::Shape padding_shape;
|
|
|
|
CUDANet::Shape out_shape;
|
|
CUDANet::Tensor output;
|
|
|
|
CUDANet::Backend *backend;
|
|
};
|
|
|
|
} // namespace CUDANet::Layers
|