training_arena_plan.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/core/memory_pool.h"
 
namespace opennn
{
 
class Loss;
struct BackPropagation;
struct ForwardPropagation;
 
// Cold metadata shared by forward and backward construction. Runtime tensor
// access never goes through this object.
class TrainingArenaPlan
{
public:
    TrainingArenaPlan(Index batch_size, Loss&, bool include_gradient);
    ~TrainingArenaPlan();
 
    TrainingArenaPlan(const TrainingArenaPlan&) = delete;
    TrainingArenaPlan& operator=(const TrainingArenaPlan&) = delete;
 
private:
    struct Impl;
    unique_ptr<Impl> impl;
 
    span<const MemoryPoolEntry> co_planned_lifetimes() const noexcept;
    bool uses_joint_gradient() const noexcept;
    void bind_offsets(span<const Index>);
 
    friend struct BackPropagation;
    friend struct ForwardPropagation;
};
 
}