| Package | Description |
|---|---|
| org.bytedeco.tvm | |
| org.bytedeco.tvm.global |
| Modifier and Type | Method and Description |
|---|---|
DLDevice |
TVMPODValue_.asDLDevice() |
DLDevice |
DLDevice.device_id(int setter) |
DLDevice |
DLDevice.device_type(int setter) |
DLDevice |
DLTensor.device()
\brief The device of the tensor
|
DLDevice |
DLDevice.getPointer(long i) |
DLDevice |
DLDevice.position(long position) |
DLDevice |
TVMValue.v_device() |
| Modifier and Type | Method and Description |
|---|---|
static boolean |
NDArray.AbilityOfZeroCopyForDLTensor(DLTensor tensor,
DLDevice dev)
\brief Check conditions for construction NDArray over DLTensor without copying.
|
void |
TVMArgsSetter.apply(long i,
DLDevice value) |
NDArray |
NDArray.CopyTo(DLDevice dev) |
NDArray |
NDArray.CopyTo(DLDevice dev,
TVMStringOptional mem_scope)
\brief Copy the data to another device.
|
DLTensor |
DLTensor.device(DLDevice setter) |
static NDArray |
NDArray.Empty(ShapeTuple shape,
DLDataType dtype,
DLDevice dev) |
static NDArray |
NDArray.Empty(ShapeTuple shape,
DLDataType dtype,
DLDevice dev,
TVMStringOptional mem_scope)
\brief Create an empty NDArray.
|
static NDArray |
NDArray.NewFromDLTensor(DLTensor dl_tensor,
DLDevice dev)
\brief Create new NDArray, data is copied from DLTensor.
|
TVMRetValue |
TVMRetValue.put(DLDevice value) |
TVMValue |
TVMValue.v_device(DLDevice setter) |
| Constructor and Description |
|---|
Container(Pointer data,
ShapeTuple shape,
DLDataType dtype,
DLDevice dev) |
| Modifier and Type | Method and Description |
|---|---|
static DLDevice |
tvm_runtime.GetPreferredHostDevice(DLDevice device)
\brief Get the preferred host device from the input device.
|
| Modifier and Type | Method and Description |
|---|---|
static DLDevice |
tvm_runtime.GetPreferredHostDevice(DLDevice device)
\brief Get the preferred host device from the input device.
|
static int |
tvm_runtime.TVMDeviceAllocDataSpace(DLDevice dev,
long nbytes,
long alignment,
DLDataType type_hint,
Pointer out_data) |
static int |
tvm_runtime.TVMDeviceAllocDataSpace(DLDevice dev,
long nbytes,
long alignment,
DLDataType type_hint,
PointerPointer out_data)
\brief Allocate a data space on device.
|
static int |
tvm_runtime.TVMDeviceAllocDataSpaceWithScope(DLDevice dev,
int ndim,
long[] shape,
DLDataType dtype,
BytePointer mem_scope,
Pointer out_data) |
static int |
tvm_runtime.TVMDeviceAllocDataSpaceWithScope(DLDevice dev,
int ndim,
long[] shape,
DLDataType dtype,
String mem_scope,
Pointer out_data) |
static int |
tvm_runtime.TVMDeviceAllocDataSpaceWithScope(DLDevice dev,
int ndim,
LongBuffer shape,
DLDataType dtype,
BytePointer mem_scope,
Pointer out_data) |
static int |
tvm_runtime.TVMDeviceAllocDataSpaceWithScope(DLDevice dev,
int ndim,
LongBuffer shape,
DLDataType dtype,
String mem_scope,
Pointer out_data) |
static int |
tvm_runtime.TVMDeviceAllocDataSpaceWithScope(DLDevice dev,
int ndim,
LongPointer shape,
DLDataType dtype,
BytePointer mem_scope,
Pointer out_data) |
static int |
tvm_runtime.TVMDeviceAllocDataSpaceWithScope(DLDevice dev,
int ndim,
LongPointer shape,
DLDataType dtype,
BytePointer mem_scope,
PointerPointer out_data)
\brief Allocate a data space on device with special memory scope.
|
static int |
tvm_runtime.TVMDeviceAllocDataSpaceWithScope(DLDevice dev,
int ndim,
LongPointer shape,
DLDataType dtype,
String mem_scope,
Pointer out_data) |
static int |
tvm_runtime.TVMDeviceFreeDataSpace(DLDevice dev,
Pointer ptr)
\brief Free a data space on device.
|
Copyright © 2024. All rights reserved.