| Package | Description |
|---|---|
| org.bytedeco.tvm | |
| org.bytedeco.tvm.global |
| Modifier and Type | Method and Description |
|---|---|
static void |
NDArray.CopyFromTo(DLTensor from,
DLTensor to,
TVMStreamHandle stream)
\brief Function to copy data from one array to another.
|
| Modifier and Type | Method and Description |
|---|---|
static int |
tvm_runtime.TVMArrayCopyFromTo(DLTensor from,
DLTensor to,
TVMStreamHandle stream)
\brief Copy the array, both from and to must be valid during the copy.
|
static int |
tvm_runtime.TVMDeviceCopyDataFromTo(DLTensor from,
DLTensor to,
TVMStreamHandle stream)
\brief Copy data from one place to another.
|
static int |
tvm_runtime.TVMSetStream(int device_type,
int device_id,
TVMStreamHandle handle)
\brief Set the runtime stream of current thread to be stream.
|
static int |
tvm_runtime.TVMStreamCreate(int device_type,
int device_id,
TVMStreamHandle out)
\brief Create a new runtime stream.
|
static int |
tvm_runtime.TVMStreamFree(int device_type,
int device_id,
TVMStreamHandle stream)
\brief Free a created stream handle.
|
static int |
tvm_runtime.TVMStreamStreamSynchronize(int device_type,
int device_id,
TVMStreamHandle src,
TVMStreamHandle dst)
\brief Synchronize two streams of execution.
|
static int |
tvm_runtime.TVMSynchronize(int device_type,
int device_id,
TVMStreamHandle stream)
\brief Wait until all computations on stream completes.
|
Copyright © 2024. All rights reserved.