|
tvm
|
TIR expressions. More...
#include <tvm/ffi/container/array.h>#include <tvm/ffi/container/map.h>#include <tvm/ffi/string.h>#include <tvm/ir/expr.h>#include <tvm/node/functor.h>#include <tvm/node/node.h>#include <tvm/runtime/base.h>#include <tvm/runtime/data_type.h>#include <tvm/tir/buffer.h>#include <tvm/tir/var.h>#include <algorithm>#include <iostream>#include <limits>#include <string>#include <unordered_map>#include <utility>Go to the source code of this file.
Classes | |
| class | tvm::tir::StringImmNode |
| ffi::String constants, only used in asserts. More... | |
| class | tvm::tir::StringImm |
| Managed reference to StringImmNode. More... | |
| class | tvm::tir::CastNode |
| Cast value from one data type to another. More... | |
| class | tvm::tir::Cast |
| Managed reference to CastNode. More... | |
| class | tvm::tir::BinaryOpNode< T > |
| Base template to implement binary ops. More... | |
| class | tvm::tir::AddNode |
| a + b More... | |
| class | tvm::tir::Add |
| Managed reference to AddNode. More... | |
| class | tvm::tir::SubNode |
| a - b More... | |
| class | tvm::tir::Sub |
| Managed reference to SubNode. More... | |
| class | tvm::tir::MulNode |
| a * b More... | |
| class | tvm::tir::Mul |
| Managed reference to MulNode. More... | |
| class | tvm::tir::DivNode |
| a / b in the C semnatics. More... | |
| class | tvm::tir::Div |
| Managed reference to DivNode. More... | |
| class | tvm::tir::ModNode |
| a % b in the C semnatics. More... | |
| class | tvm::tir::Mod |
| Managed reference to ModNode. More... | |
| class | tvm::tir::FloorDivNode |
| Floor division, floor(a/b) More... | |
| class | tvm::tir::FloorDiv |
| Managed reference to FloorDivNode. More... | |
| class | tvm::tir::FloorModNode |
| The remainder of the floordiv. More... | |
| class | tvm::tir::FloorMod |
| Managed reference to FloorModNode. More... | |
| class | tvm::tir::MinNode |
| min(a, b) More... | |
| class | tvm::tir::Min |
| Managed reference to MinNode. More... | |
| class | tvm::tir::MaxNode |
| max(a, b) More... | |
| class | tvm::tir::Max |
| Managed reference to MaxNode. More... | |
| class | tvm::tir::CmpOpNode< T > |
| Base template to implement comparison ops. More... | |
| class | tvm::tir::EQNode |
| a == b More... | |
| class | tvm::tir::EQ |
| Managed reference to EQNode. More... | |
| class | tvm::tir::NENode |
| a != b More... | |
| class | tvm::tir::NE |
| Managed reference to NENode. More... | |
| class | tvm::tir::LTNode |
| a < b More... | |
| class | tvm::tir::LT |
| Managed reference to LTNode. More... | |
| struct | tvm::tir::LENode |
| a <= b More... | |
| class | tvm::tir::LE |
| Managed reference to LENode. More... | |
| class | tvm::tir::GTNode |
| a > b More... | |
| class | tvm::tir::GT |
| Managed reference to GTNode. More... | |
| class | tvm::tir::GENode |
| a >= b More... | |
| class | tvm::tir::GE |
| Managed reference to GENode. More... | |
| class | tvm::tir::AndNode |
| a && b More... | |
| class | tvm::tir::And |
| Managed reference to AndNode. More... | |
| class | tvm::tir::OrNode |
| a || b More... | |
| class | tvm::tir::Or |
| Managed reference to OrNode. More... | |
| class | tvm::tir::NotNode |
| !a More... | |
| class | tvm::tir::Not |
| Managed reference to NotNode. More... | |
| class | tvm::tir::SelectNode |
| return true_value if condition is true, otherwise return false_value. More... | |
| class | tvm::tir::Select |
| Managed reference to SelectNode. More... | |
| class | tvm::tir::BufferLoadNode |
| Load value from the high dimension buffer. More... | |
| class | tvm::tir::BufferLoad |
| Managed reference to BufferLoadNode. More... | |
| class | tvm::tir::ProducerLoadNode |
| Load value from the result produced by the producer. More... | |
| class | tvm::tir::ProducerLoad |
| Managed reference to ProducerLoadNode. More... | |
| class | tvm::tir::RampNode |
| Construct a vector with lanes elements where its i-th element equals base + i * stride. This is useful to construct a index for a continuous vector load. More... | |
| class | tvm::tir::Ramp |
| Managed reference to RampNode. More... | |
| class | tvm::tir::BroadcastNode |
| Create a vector where all the elements are value. More... | |
| class | tvm::tir::Broadcast |
| Managed reference to BroadcastNode. More... | |
| class | tvm::tir::LetNode |
| Let binding. Bind var to value then evaluate body. More... | |
| class | tvm::tir::Let |
| Managed reference to LetNode. More... | |
| class | tvm::tir::CallNode |
| Call node. More... | |
| class | tvm::tir::Call |
| Managed reference to CallNode. More... | |
| class | tvm::tir::ShuffleNode |
| Shuffle instruction. vec = concat(vectors) result = (vec[indices[0]], vec[indices[1]] ...) More... | |
| class | tvm::tir::Shuffle |
| Managed reference to ShuffleNode. More... | |
| class | tvm::tir::CommReducerNode |
| A commutative reducer node to represent a commutative binary operator with identity element. More... | |
| class | tvm::tir::CommReducer |
| Managed reference to CommReducerNode. More... | |
| class | tvm::tir::ReduceNode |
| Reduction operator. More... | |
| class | tvm::tir::Reduce |
| Managed reference to ReduceNode. More... | |
| struct | tvm::ffi::TypeTraits< tvm::tir::StringImm > |
Namespaces | |
| tvm | |
| Performance counters for profiling via the PAPI library. | |
| tvm::tir | |
| tvm::ffi | |
Typedefs | |
| using | tvm::tir::IntImmNode = tvm::IntImmNode |
| using | tvm::tir::FloatImmNode = tvm::FloatImmNode |
Functions | |
| template<typename K , typename V > | |
| std::unordered_map< K, V > | tvm::tir::as_unordered_map (const ffi::Map< K, V > &dmap) |
TIR expressions.