type_functor.h File Reference

Functors and visitors for Relax type nodes. More...

#include <tvm/ir/node_functor.h>
#include <tvm/relax/distributed/type.h>
#include <tvm/relax/type.h>
#include <utility>

Include dependency graph for type_functor.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::relax::TypeFunctor< R(const Type &n, Args...)>
class	tvm::relax::TypeVisitor
	A type visitor. More...
class	tvm::relax::TypeMutator
	TypeMutator that mutates Relax type nodes. More...

Namespaces
	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
	tvm::relax

Macros
#define	RELAX_TYPE_FUNCTOR_DEFAULT
#define	TVM_RELAX_TYPE_FUNCTOR_DISPATCH(OP)

Detailed Description

Functors and visitors for Relax type nodes.

Macro Definition Documentation

RELAX_TYPE_FUNCTOR_DEFAULT

#define RELAX_TYPE_FUNCTOR_DEFAULT

Value:

  {                                                            \
    return VisitTypeDefault_(op, std::forward<Args>(args)...); \
  }

TVM_RELAX_TYPE_FUNCTOR_DISPATCH

#define TVM_RELAX_TYPE_FUNCTOR_DISPATCH

(

OP

)

Value:

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