|
tvm
|
Classes | |
| class | FunctionFrame |
| Managed reference to FunctionFrameObj. More... | |
| class | FunctionFrameObj |
| A frame that represents the Function containing TIR statements. More... | |
Functions | |
| FunctionFrame | Function (bool is_private, bool persistent=false) |
| The primitive function statement. | |
| FunctionFrame | DeclFunction (bool is_private=false, bool persistent=false) |
| Construct a bodyless function signature using the ordinary signature operations. | |
| Var | Arg (ffi::String name, Var var) |
| The Function variable arguments adding function. | |
| TensorVar | Arg (ffi::String name, TensorVar tensor) |
| The Function tensor arguments adding function. | |
| void | FuncName (ffi::String name) |
| The Function naming statement. | |
| void | FuncAttrs (ffi::Map< ffi::String, ffi::Any > attrs) |
| The Function annotation statement. | |
| Type | FuncRet (Type ret_type) |
| The Function return type statement. | |
| tvm::Stmt | TensorStore (Expr dest, ffi::Array< PrimExpr > indices, PrimExpr value) |
| Store data in a tensor destination. | |
| Var | Handle (ffi::Optional< PrimType > dtype=std::nullopt, ffi::String storage_scope="global") |
| Create a TIR var that represents a pointer. | |
| Var | TensorMap () |
| PrimExpr | BFloat8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat8x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat8x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat8x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat8x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat8x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat8x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat16x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat16x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat16x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat16x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat16x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat16x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat32x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat32x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat32x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat32x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat32x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat32x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat64x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat64x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat64x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat64x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat64x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | BFloat64x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float16x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float16x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float16x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float16x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float16x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float16x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float32x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float32x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float32x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float32x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float32x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float32x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float64x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float64x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float64x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float64x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float64x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float64x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt8x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt8x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt8x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt8x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt8x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt8x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt16x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt16x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt16x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt16x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt16x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt16x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt32x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt32x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt32x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt32x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt32x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt32x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt64x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt64x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt64x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt64x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt64x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | UInt64x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int8x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int8x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int8x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int8x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int8x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int8x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int16x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int16x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int16x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int16x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int16x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int16x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int32x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int32x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int32x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int32x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int32x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int32x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int64x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int64x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int64x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int64x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int64x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Int64x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E3M4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E3M4x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E3M4x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E3M4x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E3M4x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E3M4x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E3M4x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3B11FNUZ (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3B11FNUZx2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3B11FNUZx4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3B11FNUZx8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3B11FNUZx16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3B11FNUZx32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3B11FNUZx64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FN (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNx2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNx4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNx8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNx16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNx32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNx64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNUZ (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNUZx2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNUZx4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNUZx8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNUZx16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNUZx32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E4M3FNUZx64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2x2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2x4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2x8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2x16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2x32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2x64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2FNUZ (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2FNUZx2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2FNUZx4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2FNUZx8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2FNUZx16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2FNUZx32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E5M2FNUZx64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E8M0FNU (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E8M0FNUx2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E8M0FNUx4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E8M0FNUx8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E8M0FNUx16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E8M0FNUx32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float8E8M0FNUx64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E2M3FN (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E2M3FNx2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E2M3FNx4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E2M3FNx8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E2M3FNx16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E2M3FNx32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E2M3FNx64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E3M2FN (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E3M2FNx2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E3M2FNx4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E3M2FNx8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E3M2FNx16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E3M2FNx32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float6E3M2FNx64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float4E2M1FN (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float4E2M1FNx2 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float4E2M1FNx4 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float4E2M1FNx8 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float4E2M1FNx16 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float4E2M1FNx32 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Float4E2M1FNx64 (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Boolean (ffi::Optional< PrimExpr > expr=std::nullopt) |
| PrimExpr | Void (ffi::Optional< PrimExpr > expr=std::nullopt) |
The Function tensor arguments adding function.
| name | The name of the tensor. |
| tensor | The tensor argument. |
The Function variable arguments adding function.
| name | The name of the variable. |
| var | The variable argument. |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
| FunctionFrame tvm::script::ir_builder::tirx::DeclFunction | ( | bool | is_private = false, |
| bool | persistent = false |
||
| ) |
Construct a bodyless function signature using the ordinary signature operations.
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
| void tvm::script::ir_builder::tirx::FuncAttrs | ( | ffi::Map< ffi::String, ffi::Any > | attrs | ) |
The Function annotation statement.
| attrs | The annotations of the Function. |
| void tvm::script::ir_builder::tirx::FuncName | ( | ffi::String | name | ) |
The Function naming statement.
| name | The name of the Function. |
The Function return type statement.
| ret_type | The return type of the Function. |
| FunctionFrame tvm::script::ir_builder::tirx::Function | ( | bool | is_private, |
| bool | persistent = false |
||
| ) |
The primitive function statement.
|
inline |
Create a TIR var that represents a pointer.
| dtype | The optional data type of the pointer. If omitted, construct an opaque handle. |
| storage_scope | The storage scope of the pointer. |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
| tvm::Stmt tvm::script::ir_builder::tirx::TensorStore | ( | Expr | dest, |
| ffi::Array< PrimExpr > | indices, | ||
| PrimExpr | value | ||
| ) |
Store data in a tensor destination.
| dest | The destination expression. |
| indices | The indices location to be stored. |
| value | The value to be stored. |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |