dataflow_pattern_functor.h File Reference

Functors and visitors for dataflow patterns. More...

#include <tvm/relax/dataflow_pattern.h>
#include <unordered_set>
#include <utility>

Include dependency graph for dataflow_pattern_functor.h:

Go to the source code of this file.

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

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::relax

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

Detailed Description

Functors and visitors for dataflow patterns.

Macro Definition Documentation

DFPATTERN_FUNCTOR_DEFAULT

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

RELAX_DFPATTERN_FUNCTOR_DISPATCH

#define RELAX_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)...); \
  });