struct_info_functor.h File Reference

Functors and visitors for struct info. More...

#include <tvm/node/functor.h>
#include <tvm/relax/distributed/struct_info.h>
#include <tvm/relax/struct_info.h>
#include <utility>

Include dependency graph for struct_info_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::StructInfoFunctor< R(const StructInfo &n, Args...)>
class	tvm::relax::StructInfoVisitor
	A struct info visitor. More...
class	tvm::relax::StructInfoMutator
	StructInfoMutator that mutates struct info. More...

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

Macros
#define	STRUCT_INFO_FUNCTOR_DEFAULT    { return VisitStructInfoDefault_(op, std::forward<Args>(args)...); }
#define	TVM_STRUCT_INFO_FUNCTOR_DISPATCH(OP)

Detailed Description

Functors and visitors for struct info.

Macro Definition Documentation

STRUCT_INFO_FUNCTOR_DEFAULT

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

TVM_STRUCT_INFO_FUNCTOR_DISPATCH

#define TVM_STRUCT_INFO_FUNCTOR_DISPATCH

(

OP

)

Value:

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