src/modules/nn/sequential.cc¶

// Copyright 2026 The Tenzr Project Authors
//
// SPDX-License-Identifier: Apache-2.0
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "tenzr/modules/nn/sequential.h"

#include <string>

#include "tenzr/common/error.h"

namespace tenzr::nn {

  Sequential::Sequential(std::initializer_list<std::shared_ptr<Module>> modules) {
    size_t idx = 0;
    for (const auto& mod : modules) {
      Add(std::to_string(idx++), mod);
    }
  }

  void Sequential::Add(std::shared_ptr<Module> module) {
    TENZR_CHECK(module != nullptr, "Cannot add null module to Sequential");
    std::string name = std::to_string(layers_.size());
    Add(name, std::move(module));
  }

  void Sequential::Add(std::string name, std::shared_ptr<Module> module) {
    TENZR_CHECK(module != nullptr, "Cannot add null module to Sequential");
    RegisterSubmodule(name, module);
    layers_.push_back(std::move(module));
  }

  Tensor Sequential::Forward(const Tensor& input) {
    TENZR_CHECK(input.IsDefined(), "Input tensor is uninitialized");
    Tensor current = input;
    for (auto& layer : layers_) {
      current = layer->Forward(current);
    }
    return current;
  }

}  // namespace tenzr::nn