tvm
Classes | Namespaces | Macros
dataflow_pattern_functor.h File Reference

A set of passes for operating on pattern graphs. More...

#include <tvm/relay/dataflow_pattern.h>
#include <unordered_set>
#include <utility>
Include dependency graph for dataflow_pattern_functor.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  tvm::relay::DFPatternFunctor< R(const DFPattern &n, Args...)>
 
class  tvm::relay::DFPatternVisitor
 A simple visitor wrapper around DFPatternFunctor. Recursively visit the content. More...
 

Namespaces

 tvm
 runtime implementation for LibTorch/TorchScript.
 
 tvm::relay
 Relay: a high level functional IR for TVM.
 

Macros

#define DFPATTERN_FUNCTOR_DEFAULT    { return VisitDFPatternDefault_(op, std::forward<Args>(args)...); }
 
#define RELAY_DFPATTERN_FUNCTOR_DISPATCH(OP)
 

Detailed Description

A set of passes for operating on pattern graphs.

Macro Definition Documentation

◆ DFPATTERN_FUNCTOR_DEFAULT

#define DFPATTERN_FUNCTOR_DEFAULT    { return VisitDFPatternDefault_(op, std::forward<Args>(args)...); }

◆ RELAY_DFPATTERN_FUNCTOR_DISPATCH

#define RELAY_DFPATTERN_FUNCTOR_DISPATCH (   OP)
Value:
vtable.template set_dispatch<OP>([](const ObjectRef& n, TSelf* self, Args... args) { \
return self->VisitDFPattern_(static_cast<const OP*>(n.get()), std::forward<Args>(args)...); \
});