mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 14:24:22 +00:00
56 lines
1.1 KiB
C++
56 lines
1.1 KiB
C++
#pragma once
|
|
|
|
#include "layer.hpp"
|
|
|
|
namespace CUDANet::Layers {
|
|
|
|
class AvgPool2d : public Layer {
|
|
public:
|
|
AvgPool2d(
|
|
CUDANet::Shape input_shape,
|
|
CUDANet::Shape pool_shape,
|
|
CUDANet::Shape stride_shape,
|
|
CUDANet::Shape padding_shape,
|
|
CUDANet::Backend *backend
|
|
);
|
|
|
|
~AvgPool2d();
|
|
|
|
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;
|
|
|
|
protected:
|
|
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;
|
|
};
|
|
|
|
class AdaptiveAvgPool2d : public AvgPool2d {
|
|
public:
|
|
AdaptiveAvgPool2d(CUDANet::Shape input_shape, CUDANet::Shape output_shape, CUDANet::Backend *backend);
|
|
};
|
|
|
|
} // namespace CUDANet::Layers
|