TensorMath.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2015 - 2024 Intel Corporation
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef TENSORMATH_HPP_INCLUDE
#define TENSORMATH_HPP_INCLUDE
 
#include <memory>
#include <vector>
 
namespace geopm
{
    class TensorOneD;
    class TensorTwoD;
 
    class TensorMath
    {
        public:
            static std::shared_ptr<TensorMath> make_shared();
 
            virtual ~TensorMath() = default;
 
            virtual TensorOneD add(const TensorOneD &tensor_a, const TensorOneD &tensor_b) const = 0;
            virtual TensorOneD subtract(const TensorOneD &tensor_a, const TensorOneD &tensor_b) const = 0;
            virtual double inner_product(const TensorOneD &tensor_a, const TensorOneD &tensor_b) const = 0;
 
            virtual TensorOneD sigmoid(const TensorOneD &tensor) const = 0;
 
            virtual TensorOneD multiply(const TensorTwoD &, const TensorOneD &) const = 0;
    };
 
    class TensorMathImp : public TensorMath
    {
        public:
            TensorMathImp() = default;
            virtual ~TensorMathImp() = default;
            TensorOneD add(const TensorOneD &tensor_a, const TensorOneD &tensor_b) const override;
            TensorOneD subtract(const TensorOneD &tensor_a, const TensorOneD &tensor_b) const override;
            double inner_product(const TensorOneD &tensor_a, const TensorOneD &tensor_b) const override;
            TensorOneD sigmoid(const TensorOneD &tensor) const override;
            TensorOneD multiply(const TensorTwoD &, const TensorOneD &) const override;
    };
}
#endif /* TENSORMATH_HPP_INCLUDE */