tvm
Loading...
Searching...
No Matches
Namespaces | Classes | Typedefs | Functions
tvm::prim Namespace Reference

Namespaces

namespace  builtin
 

Classes

class  Add
 Managed reference to AddNode. More...
 
class  AddNode
 a + b More...
 
class  And
 Managed reference to AndNode. More...
 
class  AndNode
 a && b More...
 
class  BinaryOpNode
 Base template to implement binary ops. More...
 
class  Broadcast
 Managed reference to BroadcastNode. More...
 
class  BroadcastNode
 Create a vector where all the elements are value. More...
 
class  Cast
 Managed reference to CastNode. More...
 
class  CastNode
 Cast value from one data type to another. More...
 
class  CmpOpNode
 Base template to implement comparison ops. More...
 
class  Div
 Managed reference to DivNode. More...
 
class  DivNode
 a / b in the C semnatics. More...
 
class  EQ
 Managed reference to EQNode. More...
 
class  EQNode
 a == b More...
 
class  FloorDiv
 Managed reference to FloorDivNode. More...
 
class  FloorDivNode
 Floor division, floor(a/b) More...
 
class  FloorMod
 Managed reference to FloorModNode. More...
 
class  FloorModNode
 The remainder of the floordiv. More...
 
class  GE
 Managed reference to GENode. More...
 
class  GENode
 a >= b More...
 
class  GT
 Managed reference to GTNode. More...
 
class  GTNode
 a > b More...
 
class  LE
 Managed reference to LENode. More...
 
struct  LENode
 a <= b More...
 
class  Let
 Managed reference to LetNode. More...
 
class  LetNode
 Let binding. Bind var to value then evaluate body. More...
 
class  LT
 Managed reference to LTNode. More...
 
class  LTNode
 a < b More...
 
class  Max
 Managed reference to MaxNode. More...
 
class  MaxNode
 max(a, b) More...
 
class  Min
 Managed reference to MinNode. More...
 
class  MinNode
 min(a, b) More...
 
class  Mod
 Managed reference to ModNode. More...
 
class  ModNode
 a % b in the C semnatics. More...
 
class  Mul
 Managed reference to MulNode. More...
 
class  MulNode
 a * b More...
 
class  NE
 Managed reference to NENode. More...
 
class  NENode
 a != b More...
 
class  Not
 Managed reference to NotNode. More...
 
class  NotNode
 !a More...
 
class  Or
 Managed reference to OrNode. More...
 
class  OrNode
 a || b More...
 
class  Ramp
 Managed reference to RampNode. More...
 
class  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  Select
 Managed reference to SelectNode. More...
 
class  SelectNode
 return true_value if condition is true, otherwise return false_value. More...
 
class  Shuffle
 Managed reference to ShuffleNode. More...
 
class  ShuffleNode
 Shuffle instruction. vec = concat(vectors) result = (vec[indices[0]], vec[indices[1]] ...) More...
 
class  StringImm
 Managed reference to StringImmNode. More...
 
class  StringImmNode
 ffi::String constants, only used in asserts. More...
 
class  Sub
 Managed reference to SubNode. More...
 
class  SubNode
 a - b More...
 

Typedefs

using IntImmNode = tvm::IntImmNode
 
using FloatImmNode = tvm::FloatImmNode
 

Functions

template<typename K , typename V >
std::unordered_map< K, Vas_unordered_map (const ffi::Map< K, V > &dmap)
 

Typedef Documentation

◆ FloatImmNode

◆ IntImmNode

Function Documentation

◆ as_unordered_map()

template<typename K , typename V >
std::unordered_map< K, V > tvm::prim::as_unordered_map ( const ffi::Map< K, V > &  dmap)
inline