tvm
Classes | Namespaces | Variables
function.h File Reference

Relay Function. More...

#include <tvm/ir/function.h>
#include <tvm/relay/expr.h>
#include <string>
Include dependency graph for function.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  tvm::relay::FunctionNode
 Relay Function container. More...
 
class  tvm::relay::Function
 Managed reference to FunctionNode. More...
 

Namespaces

 tvm
 Performance counters for profiling via the PAPI library.
 
 tvm::relay
 Relay: a high level functional IR for TVM.
 
 tvm::relay::attr
 namespace of the attributes that can be attached to a relay::Function.
 

Variables

constexpr const char * tvm::relay::attr::kPrimitive = "Primitive"
 Mark the function as a primitive function. More...
 
constexpr const char * tvm::relay::attr::kCompiler = "Compiler"
 Indicate the compiler that should be used for building this function. When this is unset or set to "default", the default compilation pipeline will be used. More...
 
constexpr const char * tvm::relay::attr::kClosure = "Closure"
 Indicate if the function is a closure. More...
 
constexpr const char * tvm::relay::attr::kParams = "__params__"
 Store a Var to parameter/Constant mapping on a Function. More...
 
constexpr const char * tvm::relay::attr::kExternalSymbol = "ExternalSymbol"
 Store the unique external symbol for external compilers. More...
 
constexpr const char * tvm::relay::attr::kSkipOptimization = "SkipOptimization"
 Mark if the function should be avoided being optimized. More...
 
constexpr const char * tvm::relay::attr::kComposite = "Composite"
 Treat the function as a composite operator. More...
 
constexpr const char * tvm::relay::attr::kInline = "Inline"
 Mark the function to be inlined. More...
 
constexpr const char * tvm::relay::attr::kPartitionedFromPattern = "PartitionedFromPattern"
 Indicate the function was created by the Pattern Partitioning Pass. More...
 
constexpr const char * tvm::relay::attr::kReshapeOnly = "relay.reshape_only"
 Mark the function as only composed of reshape operations. More...
 

Detailed Description

Relay Function.