|
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/runtime/base.h>#include <tvm/runtime/data_type.h>#include <tvm/tirx/buffer.h>#include <tvm/tirx/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::tirx::StringImmNode |
| ffi::String constants, only used in asserts. More... | |
| class | tvm::tirx::StringImm |
| Managed reference to StringImmNode. More... | |
| class | tvm::tirx::CastNode |
| Cast value from one data type to another. More... | |
| class | tvm::tirx::Cast |
| Managed reference to CastNode. More... | |
| class | tvm::tirx::BinaryOpNode< T > |
| Base template to implement binary ops. More... | |
| class | tvm::tirx::AddNode |
| a + b More... | |
| class | tvm::tirx::Add |
| Managed reference to AddNode. More... | |
| class | tvm::tirx::SubNode |
| a - b More... | |
| class | tvm::tirx::Sub |
| Managed reference to SubNode. More... | |
| class | tvm::tirx::MulNode |
| a * b More... | |
| class | tvm::tirx::Mul |
| Managed reference to MulNode. More... | |
| class | tvm::tirx::DivNode |
| a / b in the C semnatics. More... | |
| class | tvm::tirx::Div |
| Managed reference to DivNode. More... | |
| class | tvm::tirx::ModNode |
| a % b in the C semnatics. More... | |
| class | tvm::tirx::Mod |
| Managed reference to ModNode. More... | |
| class | tvm::tirx::FloorDivNode |
| Floor division, floor(a/b) More... | |
| class | tvm::tirx::FloorDiv |
| Managed reference to FloorDivNode. More... | |
| class | tvm::tirx::FloorModNode |
| The remainder of the floordiv. More... | |
| class | tvm::tirx::FloorMod |
| Managed reference to FloorModNode. More... | |
| class | tvm::tirx::MinNode |
| min(a, b) More... | |
| class | tvm::tirx::Min |
| Managed reference to MinNode. More... | |
| class | tvm::tirx::MaxNode |
| max(a, b) More... | |
| class | tvm::tirx::Max |
| Managed reference to MaxNode. More... | |
| class | tvm::tirx::CmpOpNode< T > |
| Base template to implement comparison ops. More... | |
| class | tvm::tirx::EQNode |
| a == b More... | |
| class | tvm::tirx::EQ |
| Managed reference to EQNode. More... | |
| class | tvm::tirx::NENode |
| a != b More... | |
| class | tvm::tirx::NE |
| Managed reference to NENode. More... | |
| class | tvm::tirx::LTNode |
| a < b More... | |
| class | tvm::tirx::LT |
| Managed reference to LTNode. More... | |
| struct | tvm::tirx::LENode |
| a <= b More... | |
| class | tvm::tirx::LE |
| Managed reference to LENode. More... | |
| class | tvm::tirx::GTNode |
| a > b More... | |
| class | tvm::tirx::GT |
| Managed reference to GTNode. More... | |
| class | tvm::tirx::GENode |
| a >= b More... | |
| class | tvm::tirx::GE |
| Managed reference to GENode. More... | |
| class | tvm::tirx::AndNode |
| a && b More... | |
| class | tvm::tirx::And |
| Managed reference to AndNode. More... | |
| class | tvm::tirx::OrNode |
| a || b More... | |
| class | tvm::tirx::Or |
| Managed reference to OrNode. More... | |
| class | tvm::tirx::NotNode |
| !a More... | |
| class | tvm::tirx::Not |
| Managed reference to NotNode. More... | |
| class | tvm::tirx::SelectNode |
| return true_value if condition is true, otherwise return false_value. More... | |
| class | tvm::tirx::Select |
| Managed reference to SelectNode. More... | |
| class | tvm::tirx::BufferLoadNode |
| Load value from the high dimension buffer. More... | |
| class | tvm::tirx::BufferLoad |
| Managed reference to BufferLoadNode. More... | |
| class | tvm::tirx::ProducerLoadNode |
| Load value from the result produced by the producer. More... | |
| class | tvm::tirx::ProducerLoad |
| Managed reference to ProducerLoadNode. More... | |
| class | tvm::tirx::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::tirx::Ramp |
| Managed reference to RampNode. More... | |
| class | tvm::tirx::BroadcastNode |
| Create a vector where all the elements are value. More... | |
| class | tvm::tirx::Broadcast |
| Managed reference to BroadcastNode. More... | |
| class | tvm::tirx::LetNode |
| Let binding. Bind var to value then evaluate body. More... | |
| class | tvm::tirx::Let |
| Managed reference to LetNode. More... | |
| class | tvm::tirx::CallNode |
| Call node. More... | |
| class | tvm::tirx::Call |
| Managed reference to CallNode. More... | |
| class | tvm::tirx::ShuffleNode |
| Shuffle instruction. vec = concat(vectors) result = (vec[indices[0]], vec[indices[1]] ...) More... | |
| class | tvm::tirx::Shuffle |
| Managed reference to ShuffleNode. More... | |
| class | tvm::tirx::CommReducerNode |
| A commutative reducer node to represent a commutative binary operator with identity element. More... | |
| class | tvm::tirx::CommReducer |
| Managed reference to CommReducerNode. More... | |
| class | tvm::tirx::ReduceNode |
| Reduction operator. More... | |
| class | tvm::tirx::Reduce |
| Managed reference to ReduceNode. More... | |
| struct | tvm::ffi::TypeTraits< tvm::tirx::StringImm > |
Namespaces | |
| tvm | |
| An object that builds and maintains block scope and StmtSref mapping for Dependence analysis. | |
| tvm::tirx | |
| tvm::ffi | |
Typedefs | |
| using | tvm::tirx::IntImmNode = tvm::IntImmNode |
| using | tvm::tirx::FloatImmNode = tvm::FloatImmNode |
Functions | |
| template<typename K , typename V > | |
| std::unordered_map< K, V > | tvm::tirx::as_unordered_map (const ffi::Map< K, V > &dmap) |
TIR expressions.