(allocate-device-array shape datatype device-type device-id)
(check-cpu-tensor item)
(copy-array-to-array! src dst stream)
(copy-from-array! src-tensor dest n-bytes)
(copy-to-array! src dest-tensor n-bytes)
(ensure-array item)
(ensure-tensor item)
(pointer->tvm-ary ptr
device-type
device-id
datatype
shape
strides
byte-offset
&
[gc-root])
Not all backends in TVM can offset their pointer types. For this reason, tvm arrays have a byte_offset member that you can use to make an array not start at the pointer's base address. If provided this new object will keep the gc-root alive in the eyes of the gc.
Not all backends in TVM can offset their pointer types. For this reason, tvm arrays have a byte_offset member that you can use to make an array not start at the pointer's base address. If provided this new object will keep the gc-root alive in the eyes of the gc.
(to-size-t item)
(TVMArrayAlloc shape
n-dim
dtype_code
dtype_bits
dtype_lanes
device_type
device_id
retval)
Allocate a new tvm array
Allocate a new tvm array
(TVMArrayCopyFromBytes dest-tensor src n-bytes)
Copy bytes into an array
Copy bytes into an array
(TVMArrayCopyFromTo src dst stream)
Copy data from an array to an array
Copy data from an array to an array
(TVMArrayCopyToBytes src-tensor dest n-bytes)
Copy tensor data to bytes
Copy tensor data to bytes
(TVMArrayFree item)
Free a TVM array allocated with TVMArrayAlloc
Free a TVM array allocated with TVMArrayAlloc
cljdoc is a website building & hosting documentation for Clojure/Script libraries
× close