tvm
|
#include <expr.h>
Public Member Functions | |
void | VisitAttrs (tvm::AttrVisitor *v) |
bool | SEqualReduce (const CallNode *other, SEqualReducer equal) const |
void | SHashReduce (SHashReducer hash_reduce) const |
TVM_DECLARE_FINAL_OBJECT_INFO (CallNode, ExprNode) | |
![]() | |
const Type & | checked_type () const |
template<typename TTypeNode > | |
const TTypeNode * | type_as () const |
Check if the inferred(checked) type of the Expr is backed by a TTypeNode and return it. More... | |
VirtualDevice | virtual_device () const |
TVM_DECLARE_BASE_OBJECT_INFO (RelayExprNode, BaseExprNode) | |
![]() | |
TVM_DECLARE_BASE_OBJECT_INFO (BaseExprNode, Object) | |
![]() | |
uint32_t | type_index () const |
std::string | GetTypeKey () const |
size_t | GetTypeKeyHash () const |
template<typename TargetType > | |
bool | IsInstance () const |
bool | unique () const |
Object () | |
Object (const Object &other) | |
Object (Object &&other) | |
Object & | operator= (const Object &other) |
Object & | operator= (Object &&other) |
Public Attributes | |
Expr | op |
The operator(function) being invoked. More... | |
tvm::Array< relay::Expr > | args |
The arguments(inputs) of the call. More... | |
Attrs | attrs |
The additional attributes. More... | |
tvm::Array< Type > | type_args |
The type arguments passed to polymorphic(template) function. More... | |
![]() | |
Type | checked_type_ = Type(nullptr) |
Stores the result of type inference(type checking). More... | |
ObjectRef | virtual_device_ |
The virtual device (VirtualDevice) for this node (the result of device planning). For first-order expressions (non functions), this describes where the result of evaluating the expression should be stored. Note that currently, all composite first-order values (tuples, references, ADTs) must be stored on the same virtual device. This means that it is not possible to store two tuple fields on different devices, so we only need one virtual device for these types. More... | |
![]() | |
Span | span |
Span that points to the original source code. Reserved debug information. More... | |
Static Public Attributes | |
static constexpr const char * | _type_key = "relay.Call" |
![]() | |
static constexpr const char * | _type_key = "RelayExpr" |
static constexpr const uint32_t | _type_child_slots = 22 |
![]() | |
static constexpr const char * | _type_key = "BaseExpr" |
static constexpr const bool | _type_has_method_sequal_reduce = true |
static constexpr const bool | _type_has_method_shash_reduce = true |
static constexpr const uint32_t | _type_child_slots = 62 |
![]() | |
static constexpr const char * | _type_key = "runtime.Object" |
static constexpr bool | _type_final = false |
static constexpr uint32_t | _type_child_slots = 0 |
static constexpr bool | _type_child_slots_can_overflow = true |
static constexpr bool | _type_has_method_visit_attrs = true |
static constexpr bool | _type_has_method_sequal_reduce = false |
static constexpr bool | _type_has_method_shash_reduce = false |
static constexpr uint32_t | _type_index = TypeIndex::kDynamic |
Static Protected Member Functions | |
static void | Deleter_ (Object *ptr) |
![]() | |
static uint32_t | GetOrAllocRuntimeTypeIndex (const std::string &key, uint32_t static_tindex, uint32_t parent_tindex, uint32_t type_child_slots, bool type_child_slots_can_overflow) |
Get the type index using type key. More... | |
Protected Attributes | |
Object::FDeleter | saved_deleter_ |
![]() | |
uint32_t | type_index_ {0} |
Type index(tag) that indicates the type of the object. More... | |
RefCounterType | ref_counter_ {0} |
The internal reference counter. More... | |
FDeleter | deleter_ = nullptr |
deleter of this object to enable customized allocation. If the deleter is nullptr, no deletion will be performed. The creator of the object must always set the deleter field properly. More... | |
Friends | |
template<typename > | |
class | runtime::ObjAllocatorBase |
class | Call |
Additional Inherited Members | |
![]() | |
typedef void(* | FDeleter) (Object *self) |
Object deleter. More... | |
using | RefCounterType = std::atomic< int32_t > |
![]() | |
static std::string | TypeIndex2Key (uint32_t tindex) |
Get the type key of the corresponding index from runtime. More... | |
static size_t | TypeIndex2KeyHash (uint32_t tindex) |
Get the type key hash of the corresponding index from runtime. More... | |
static uint32_t | TypeKey2Index (const std::string &key) |
Get the type index of the corresponding key from runtime. More... | |
static uint32_t | _GetOrAllocRuntimeTypeIndex () |
static uint32_t | RuntimeTypeIndex () |
![]() | |
void | IncRef () |
developer function, increases reference counter. More... | |
void | DecRef () |
developer function, decrease reference counter. More... | |
Call container.
|
staticprotected |
|
inline |
|
inline |
|
inline |
|
friend |
|
friend |
|
staticconstexpr |
tvm::Array<relay::Expr> tvm::relay::CallNode::args |
The arguments(inputs) of the call.
Attrs tvm::relay::CallNode::attrs |
The additional attributes.
Expr tvm::relay::CallNode::op |
|
protected |
tvm::Array<Type> tvm::relay::CallNode::type_args |
The type arguments passed to polymorphic(template) function.
This is the advance feature that is only used when the function is polymorphic. It is safe to be ignored in most cases. For example, in the following code, the type_args of addone call is [int].