Node.h

Go to the documentation of this file.

/*
 * Copyright 2017-2020 CNRS-AIST JRL and CNRS-UM LIRMM
 */
 
#pragma once
 
#include <tvm/graph/internal/AbstractNode.h>
 
namespace tvm
{
 
namespace graph
{
 
namespace abstract
{
 
template<typename T>
class Node : public internal::AbstractNode
{
protected:
  template<typename EnumT, typename U, typename... Args>
  void registerUpdates(EnumT u, void (U::*fn)(), Args... args);
 
  template<typename EnumT, typename U>
  void registerUpdates(EnumT u, void (U::*fn)());
 
  template<typename U = T, typename EnumO, typename EnumU>
  void addOutputDependency(EnumO o, EnumU u);
 
  template<typename U = T, typename EnumO, typename EnumU>
  void addOutputDependency(std::initializer_list<EnumO> os, EnumU u);
 
  template<typename U = T, typename EnumU1, typename EnumU2>
  void addInternalDependency(EnumU1 uDependent, EnumU2 u);
 
  template<typename U = T, typename EnumU, typename S, typename EnumO, typename... Args>
  void addInputDependency(EnumU u, std::shared_ptr<S> source, EnumO i, Args... args);
 
  template<typename U = T,
           typename EnumU,
           typename S,
           typename EnumO,
           typename... Args,
           typename std::enable_if<std::is_base_of<abstract::Outputs, S>::value, int>::type = 0>
  void addInputDependency(EnumU u, S & source, EnumO i, Args... args);
 
  template<typename U = T, typename EnumO, typename S, typename EnumI>
  void addDirectDependency(EnumO o, std::shared_ptr<S> source, EnumI i);
 
  template<typename U = T,
           typename EnumO,
           typename S,
           typename EnumI,
           typename std::enable_if<std::is_base_of<abstract::Outputs, S>::value, int>::type = 0>
  void addDirectDependency(EnumO o, S & source, EnumI i);
 
private:
  /* Internal version that takes a pointer to the source */
  template<typename U, typename EnumU, typename S>
  void checkAddInputDependency(EnumU u);
  template<typename U = T, typename EnumU, typename S, typename EnumO>
  void addInputDependency(EnumU u, S * source, EnumO i);
  template<typename U = T, typename EnumU, typename S, typename EnumO, typename... Args>
  void addInputDependency(EnumU u, S * source, EnumO i, Args... args);
  template<typename U, typename EnumO, typename S, typename EnumI>
  void checkAddDirectDependency(EnumO o);
  template<typename U = T, typename EnumO, typename S, typename EnumI>
  void addDirectDependency(EnumO o, S * source, EnumI i);
};
 
} // namespace abstract
 
} // namespace graph
 
} // namespace tvm
 
#include "Node.hpp"