tvm
|
Base expr nodes in TVM. More...
#include <tvm/ir/source_map.h>
#include <tvm/ir/type.h>
#include <tvm/node/node.h>
#include <tvm/runtime/container/string.h>
#include <tvm/runtime/object.h>
#include <algorithm>
#include <functional>
#include <limits>
#include <string>
#include <type_traits>
Go to the source code of this file.
Classes | |
class | tvm::BaseExprNode |
Base type of all the expressions. More... | |
class | tvm::BaseExpr |
Managed reference to BaseExprNode. More... | |
class | tvm::PrimExprNode |
Base node of all primitive expressions. More... | |
class | tvm::PrimExpr |
Reference to PrimExprNode. More... | |
class | tvm::RelayExprNode |
Base node of all non-primitive expressions. More... | |
class | tvm::RelayExpr |
Managed reference to RelayExprNode. More... | |
class | tvm::GlobalVarNode |
Global variable that lives in the top-level module. More... | |
class | tvm::GlobalVar |
Managed reference to GlobalVarNode. More... | |
class | tvm::IntImmNode |
Constant integer literals in the program. More... | |
class | tvm::IntImm |
Managed reference class to IntImmNode. More... | |
class | tvm::FloatImmNode |
Constant floating point literals in the program. More... | |
class | tvm::FloatImm |
Managed reference class to FloatImmNode. More... | |
class | tvm::Bool |
Boolean constant. More... | |
class | tvm::Integer |
Container of constant int that adds more constructors. More... | |
class | tvm::RangeNode |
range over one dimension More... | |
class | tvm::Range |
Range container More... | |
struct | tvm::runtime::PackedFuncValueConverter< tvm::IntImm > |
struct | tvm::runtime::PackedFuncValueConverter< tvm::Integer > |
struct | tvm::runtime::PackedFuncValueConverter< tvm::Bool > |
struct | tvm::runtime::PackedFuncValueConverter< tvm::FloatImm > |
struct | tvm::runtime::PackedFuncValueConverter< runtime::Int > |
Namespaces | |
tvm | |
runtime implementation for LibTorch/TorchScript. | |
tvm::runtime | |
Functions | |
PrimExpr | tvm::operator+ (PrimExpr a, PrimExpr b) |
add operator More... | |
PrimExpr | tvm::operator- (PrimExpr a, PrimExpr b) |
subtraction operator More... | |
PrimExpr | tvm::operator- (PrimExpr a) |
negation. More... | |
PrimExpr | tvm::operator* (PrimExpr a, PrimExpr b) |
multiplication operator More... | |
PrimExpr | tvm::operator/ (PrimExpr a, PrimExpr b) |
division operator More... | |
PrimExpr | tvm::operator<< (PrimExpr a, PrimExpr b) |
left shift operator More... | |
PrimExpr | tvm::operator>> (PrimExpr a, PrimExpr b) |
right shift operator More... | |
PrimExpr | tvm::operator> (PrimExpr a, PrimExpr b) |
greater More... | |
PrimExpr | tvm::operator>= (PrimExpr a, PrimExpr b) |
greater_equal More... | |
PrimExpr | tvm::operator< (PrimExpr a, PrimExpr b) |
less More... | |
PrimExpr | tvm::operator<= (PrimExpr a, PrimExpr b) |
less_equal More... | |
PrimExpr | tvm::operator== (PrimExpr a, PrimExpr b) |
equal More... | |
PrimExpr | tvm::operator!= (PrimExpr a, PrimExpr b) |
not_equal More... | |
PrimExpr | tvm::operator&& (PrimExpr a, PrimExpr b) |
and More... | |
PrimExpr | tvm::operator|| (PrimExpr a, PrimExpr b) |
or More... | |
PrimExpr | tvm::operator! (PrimExpr a) |
not More... | |
PrimExpr | tvm::operator& (PrimExpr a, PrimExpr b) |
take bitwise and of two values More... | |
PrimExpr | tvm::operator| (PrimExpr a, PrimExpr b) |
take bitwise or of two values More... | |
PrimExpr | tvm::operator^ (PrimExpr a, PrimExpr b) |
take bitwise xor of two values More... | |
PrimExpr | tvm::operator~ (PrimExpr a) |
take bitwise negation of two values More... | |
Bool | tvm::operator|| (const Bool &a, bool b) |
Bool | tvm::operator|| (bool a, const Bool &b) |
Bool | tvm::operator|| (const Bool &a, const Bool &b) |
Bool | tvm::operator&& (const Bool &a, bool b) |
Bool | tvm::operator&& (bool a, const Bool &b) |
Bool | tvm::operator&& (const Bool &a, const Bool &b) |
bool | tvm::operator== (const Bool &a, bool b) |
bool | tvm::operator== (bool a, const Bool &b) |
bool | tvm::operator== (const Bool &a, const Bool &b) |
Base expr nodes in TVM.