tvm
tvm::runtime::DeviceAPI Member List

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)=0tvm::runtime::DeviceAPIpure virtual
AllocDataSpace(Device dev, int ndim, const int64_t *shape, DLDataType dtype, Optional< String > mem_scope=NullOpt)tvm::runtime::DeviceAPIvirtual
AllocWorkspace(Device dev, size_t nbytes, DLDataType type_hint={})tvm::runtime::DeviceAPIvirtual
CopyDataFromTo(DLTensor *from, DLTensor *to, TVMStreamHandle stream)tvm::runtime::DeviceAPIvirtual
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::DeviceAPIprotectedvirtual
CreateStream(Device dev)tvm::runtime::DeviceAPIvirtual
FreeDataSpace(Device dev, void *ptr)=0tvm::runtime::DeviceAPIpure virtual
FreeStream(Device dev, TVMStreamHandle stream)tvm::runtime::DeviceAPIvirtual
FreeWorkspace(Device dev, void *ptr)tvm::runtime::DeviceAPIvirtual
Get(Device dev, bool allow_missing=false)tvm::runtime::DeviceAPIstatic
GetAttr(Device dev, DeviceAttrKind kind, TVMRetValue *rv)=0tvm::runtime::DeviceAPIpure virtual
GetCurrentStream(Device dev)tvm::runtime::DeviceAPIvirtual
GetDataSize(const DLTensor &arr, Optional< String > mem_scope=NullOpt)tvm::runtime::DeviceAPIvirtual
GetTargetProperty(Device dev, const std::string &property, TVMRetValue *rv)tvm::runtime::DeviceAPIinlinevirtual
NeedSetDevice(int device_type)tvm::runtime::DeviceAPIinlinestatic
SetDevice(Device dev)=0tvm::runtime::DeviceAPIpure virtual
SetStream(Device dev, TVMStreamHandle stream)tvm::runtime::DeviceAPIinlinevirtual
StreamSync(Device dev, TVMStreamHandle stream)=0tvm::runtime::DeviceAPIpure virtual
SupportsDevicePointerArithmeticsOnHost()tvm::runtime::DeviceAPIinlinevirtual
SyncStreamFromTo(Device dev, TVMStreamHandle event_src, TVMStreamHandle event_dst)tvm::runtime::DeviceAPIvirtual
~DeviceAPI()tvm::runtime::DeviceAPIinlinevirtual