load_graph.hpp

/****************************************************************************
 *  hipipe library
 *  Copyright (c) 2017, Cognexa Solutions s.r.o.
 *  Copyright (c) 2018, Iterait a.s.
 *  Author(s) Filip Matzner
 *
 *  This file is distributed under the MIT License.
 *  See the accompanying file LICENSE.txt for the complete license agreement.
 ****************************************************************************/
 
#ifndef HIPIPE_TENSORFLOW_LOAD_GRAPH_HPP
#define HIPIPE_TENSORFLOW_LOAD_GRAPH_HPP
 
#include <tensorflow/core/framework/tensor.h>
#include <tensorflow/core/public/session.h>
 
#include <experimental/filesystem>
#include <memory>
 
namespace hipipe::tensorflow {
 
std::unique_ptr<::tensorflow::Session>
load_graph(std::experimental::filesystem::path graph_file_name);
 
}  // namespace hipipe::tensorflow
#endif