pool3d_operator.h

Go to the documentation of this file.

// SPDX-License-Identifier: AGPL-3.0-only
// Copyright (C) 2026 Artificial Intelligence Techniques, S.L. and Roberto López González
 
#pragma once
 
#include "opennn/network/operators/operator.h"
 
namespace opennn
{
 
void max_pooling_3d_forward(const TensorView&, TensorView&, TensorView&, bool is_training, SequenceLengths);
void average_pooling_3d_forward(const TensorView&, TensorView&, SequenceLengths);
void max_pooling_3d_backward(const TensorView&, const TensorView&, TensorView&);
void average_pooling_3d_backward(const TensorView&, const TensorView&, TensorView&, SequenceLengths);
void first_token_3d_forward(const TensorView&, TensorView&);
void first_token_3d_backward(const TensorView&, TensorView&);
 
struct Pool3dOperator : Operator
{
    enum Method { Max, Average, First };
    Method method = Average;
 
    void forward_propagate(ForwardPropagation&, size_t, ForwardPropagationMode) override;
    void back_propagate(ForwardPropagation&, BackPropagation&, size_t) const override;
};
 
}