|
tvm
|
This is the complete list of members for tvm::runtime::DeviceAPI, including all inherited members.
| AllocDataSpace(Device dev, size_t nbytes, size_t alignment, DLDataType type_hint)=0 | tvm::runtime::DeviceAPI | pure virtual |
| AllocDataSpace(Device dev, int ndim, const int64_t *shape, DLDataType dtype, ffi::Optional< ffi::String > mem_scope=std::nullopt) | tvm::runtime::DeviceAPI | virtual |
| AllocWorkspace(Device dev, size_t nbytes, DLDataType type_hint={}) | tvm::runtime::DeviceAPI | virtual |
| CopyDataFromTo(DLTensor *from, DLTensor *to, TVMStreamHandle stream) | tvm::runtime::DeviceAPI | virtual |
| CopyDataFromTo(const void *from, size_t from_offset, void *to, size_t to_offset, size_t num_bytes, Device dev_from, Device dev_to, DLDataType type_hint, TVMStreamHandle stream) | tvm::runtime::DeviceAPI | protectedvirtual |
| CreateStream(Device dev) | tvm::runtime::DeviceAPI | virtual |
| FreeDataSpace(Device dev, void *ptr)=0 | tvm::runtime::DeviceAPI | pure virtual |
| FreeStream(Device dev, TVMStreamHandle stream) | tvm::runtime::DeviceAPI | virtual |
| FreeWorkspace(Device dev, void *ptr) | tvm::runtime::DeviceAPI | virtual |
| Get(Device dev, bool allow_missing=false) | tvm::runtime::DeviceAPI | static |
| GetAttr(Device dev, DeviceAttrKind kind, ffi::Any *rv)=0 | tvm::runtime::DeviceAPI | pure virtual |
| GetCurrentStream(Device dev) | tvm::runtime::DeviceAPI | virtual |
| GetDataSize(const DLTensor &arr, ffi::Optional< ffi::String > mem_scope=std::nullopt) | tvm::runtime::DeviceAPI | virtual |
| GetTargetProperty(Device dev, const std::string &property, ffi::Any *rv) | tvm::runtime::DeviceAPI | inlinevirtual |
| NeedSetDevice(int device_type) | tvm::runtime::DeviceAPI | inlinestatic |
| SetDevice(Device dev)=0 | tvm::runtime::DeviceAPI | pure virtual |
| SetStream(Device dev, TVMStreamHandle stream) | tvm::runtime::DeviceAPI | virtual |
| StreamSync(Device dev, TVMStreamHandle stream)=0 | tvm::runtime::DeviceAPI | pure virtual |
| SupportsDevicePointerArithmeticsOnHost() | tvm::runtime::DeviceAPI | inlinevirtual |
| SyncStreamFromTo(Device dev, TVMStreamHandle event_src, TVMStreamHandle event_dst) | tvm::runtime::DeviceAPI | virtual |
| ~DeviceAPI() | tvm::runtime::DeviceAPI | inlinevirtual |