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, Optional< String > mem_scope=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, TVMRetValue *rv)=0 | tvm::runtime::DeviceAPI | pure virtual |
GetCurrentStream(Device dev) | tvm::runtime::DeviceAPI | virtual |
GetDataSize(const DLTensor &arr, Optional< String > mem_scope=NullOpt) | tvm::runtime::DeviceAPI | virtual |
GetTargetProperty(Device dev, const std::string &property, TVMRetValue *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 | inlinevirtual |
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 |