tvm-clj.jna.dl-tensor — tvm-clj 4.3

*debug-dl-tensor-lifespan* clj

allocate-device-array clj

(allocate-device-array shape datatype device-type device-id)

check-cpu-tensor clj

(check-cpu-tensor item)

copy-array-to-array! clj

(copy-array-to-array! src dst stream)

copy-from-array! clj

(copy-from-array! src-tensor dest n-bytes)

copy-to-array! clj

(copy-to-array! src dest-tensor n-bytes)

ensure-array clj

(ensure-array item)

ensure-tensor clj

(ensure-tensor item)

pointer->tvm-ary clj

(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.

to-size-t clj

(to-size-t item)

TVMArrayAlloc clj

(TVMArrayAlloc shape
               n-dim
               dtype_code
               dtype_bits
               dtype_lanes
               device_type
               device_id
               retval)

Allocate a new tvm array

TVMArrayCopyFromBytes clj

(TVMArrayCopyFromBytes dest-tensor src n-bytes)

Copy bytes into an array

TVMArrayCopyFromTo clj

(TVMArrayCopyFromTo src dst stream)

Copy data from an array to an array

TVMArrayCopyToBytes clj

(TVMArrayCopyToBytes src-tensor dest n-bytes)

Copy tensor data to bytes

TVMArrayFree clj

(TVMArrayFree item)

Free a TVM array allocated with TVMArrayAlloc