tvm::NodeFunctor< FType > Class Template Reference

A dynamically dispatched functor on the type of the first argument. More...

Collaboration diagram for tvm::NodeFunctor< FType >:

Detailed Description

template<typename FType>
class tvm::NodeFunctor< FType >

A dynamically dispatched functor on the type of the first argument.

This is a class that is useful to construct polymorphic dispatching base on the AST/IR node's type.

NodeFunctor<std::string (const ffi::ObjectRef& n, std::string prefix)> tostr;
tostr.set_dispatch<Add>([](const ffi::ObjectRef& op, std::string prefix) {
  return prefix + "Add";
});
tostr.set_dispatch<IntImm>([](const ffi::ObjectRef& op, std::string prefix) {
  return prefix + "IntImm"
});
 
Expr x = MakeConst(1);
Expr y = x + x;
// dispatch to IntImm, outputs "MyIntImm"
LOG(INFO) << tostr(x, "My");
// dispatch to IntImm, outputs "MyAdd"
LOG(INFO) << tostr(y, "My");

Template Parameters

FType

function signiture This type if only defined for FType with function signature

---

The documentation for this class was generated from the following file:

• include/tvm/ir/node_functor.h