| Package | Description |
|---|---|
| org.bytedeco.tvm | |
| org.bytedeco.tvm.global |
| Modifier and Type | Method and Description |
|---|---|
DLContext |
TVMPODValue_.asDLContext() |
DLContext |
DLTensor.ctx()
\brief The device context of the tensor
|
DLContext |
DLContext.device_id(int setter) |
DLContext |
DLContext.device_type(int setter) |
DLContext |
DLContext.getPointer(long i) |
DLContext |
DLContext.position(long position) |
DLContext |
TVMValue.v_ctx() |
| Modifier and Type | Method and Description |
|---|---|
void |
TVMArgsSetter.apply(long i,
DLContext value) |
NDArray |
NDArray.CopyTo(DLContext ctx)
\brief Copy the data to another context.
|
DLTensor |
DLTensor.ctx(DLContext setter) |
static NDArray |
NDArray.Empty(long[] shape,
DLDataType dtype,
DLContext ctx) |
static NDArray |
NDArray.Empty(LongBuffer shape,
DLDataType dtype,
DLContext ctx) |
static NDArray |
NDArray.Empty(LongPointer shape,
DLDataType dtype,
DLContext ctx)
\brief Create an empty NDArray.
|
TVMRetValue |
TVMRetValue.put(DLContext value) |
TVMValue |
TVMValue.v_ctx(DLContext setter) |
| Constructor and Description |
|---|
Container(Pointer data,
long[] shape,
DLDataType dtype,
DLContext ctx) |
Container(Pointer data,
LongBuffer shape,
DLDataType dtype,
DLContext ctx) |
Container(Pointer data,
LongPointer shape,
DLDataType dtype,
DLContext ctx) |
| Modifier and Type | Method and Description |
|---|---|
static int |
tvm_runtime.TVMDeviceAllocDataSpace(DLContext ctx,
long nbytes,
long alignment,
DLDataType type_hint,
Pointer out_data) |
static int |
tvm_runtime.TVMDeviceAllocDataSpace(DLContext ctx,
long nbytes,
long alignment,
DLDataType type_hint,
PointerPointer out_data)
\brief Allocate a data space on device.
|
static int |
tvm_runtime.TVMDeviceCopyDataFromTo(Pointer from,
long from_offset,
Pointer to,
long to_offset,
long num_bytes,
DLContext ctx_from,
DLContext ctx_to,
DLDataType type_hint,
TVMStreamHandle stream)
\brief Copy data from one place to another.
|
static int |
tvm_runtime.TVMDeviceFreeDataSpace(DLContext ctx,
Pointer ptr)
\brief Free a data space on device.
|
Copyright © 2021. All rights reserved.