tvm
Classes | Namespaces | Macros | Typedefs | Functions
base.h File Reference

Base classes for the Relay IR. More...

#include <tvm/ir/source_map.h>
#include <tvm/node/node.h>
#include <tvm/tir/expr.h>
#include <string>
#include <vector>
Include dependency graph for base.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  tvm::relay::RelayNode
 This is the base node container of all relay structures. More...
 
class  tvm::relay::IdNode
 The unique identifier of variables. More...
 
class  tvm::relay::Id
 

Namespaces

 tvm
 runtime implementation for LibTorch/TorchScript.
 
 tvm::relay
 Relay: a high level functional IR for TVM.
 

Macros

#define RELAY_DEBUG(...)
 
#define RELAY_DEBUG_INTERP(...)
 

Typedefs

using tvm::relay::IndexExpr = ::tvm::PrimExpr
 Symbolic expression for tensor shape. More...
 
using tvm::relay::SourceName = tvm::SourceName
 
using tvm::relay::Span = tvm::Span
 
using tvm::relay::SpanNode = tvm::SpanNode
 

Functions

String tvm::relay::PrettyPrint (const ObjectRef &node)
 Pretty print a node for debug purposes. More...
 
String tvm::relay::AsText (const ObjectRef &node, bool show_meta_data=true, runtime::TypedPackedFunc< String(ObjectRef)> annotate=nullptr)
 Render the node as a string in the text format. More...
 

Detailed Description

Base classes for the Relay IR.

Macro Definition Documentation

◆ RELAY_DEBUG

#define RELAY_DEBUG (   ...)
Value:
{ \
auto fdebug = runtime::Registry::Get("relay.debug"); \
ICHECK(fdebug) << "Could not find Relay Python debugger function."; \
(*fdebug)("RELAY_DEBUG", __FILE__, __LINE__, __VA_ARGS__); \
}

◆ RELAY_DEBUG_INTERP

#define RELAY_DEBUG_INTERP (   ...)
Value:
{ \
auto fdebug = runtime::Registry::Get("relay.debug_interp"); \
ICHECK(fdebug) << "Could not find Relay Python debugger function."; \
(*fdebug)("RELAY_DEBUG", __FILE__, __LINE__, __VA_ARGS__); \
}