graph.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/graph/CallGraph.h>
#include <tvm/graph/internal/Inputs.h>
#include <tvm/utils/internal/graphDetails.h>
 
#include <memory>
 
namespace tvm
{
 
namespace utils
{
 
template<typename Object, typename... Args>
inline std::unique_ptr<graph::CallGraph> generateUpdateGraph(std::shared_ptr<Object> obj, Args &&... args);
 
template<typename Object, typename... Args>
inline std::unique_ptr<graph::CallGraph> generateUpdateGraph(std::shared_ptr<Object> obj, Args &&... args)
{
  auto ptr = std::unique_ptr<graph::CallGraph>(new graph::CallGraph());
  auto user = std::make_shared<graph::internal::Inputs>();
  internal::parseSourcesAndOutputs(ptr.get(), user, obj, std::forward<Args>(args)...);
  ptr->update();
 
  return ptr;
}
 
} // namespace utils
 
} // namespace tvm