tvm-clj.tvm-jna — tvm-clj 4.5

->node clj

(->node item)

->tvm clj

(->tvm item)

->tvm-value clj

(->tvm-value item)

allocate-device-array clj

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

base-ptr clj

(base-ptr item)

byte-offset clj

(byte-offset item)

call-function clj

(call-function tvm-fn & args)

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)

create-stream clj

(create-stream device-type device-id)

device-exists? clj

(device-exists? device-type device-id)

device-id clj

(device-id item)

device-type clj

(device-type item)

device-type->int clj

(device-type->int device-type)

device-type-int->device-type clj

(device-type-int->device-type device-type)

g-fn clj

get-module-function clj

(get-module-function module fn-name & {:keys [query-imports?]})

get-module-source clj

(get-module-source module {:keys [format] :or {format ""}})

get-node-type clj

(get-node-type node-handle)

global-function clj

(global-function fn-name & args)

global-node-function clj

gn-fn clj

is-expression-node? clj

(is-expression-node? node)

is-node-handle? clj

(is-node-handle? item)

mod-import clj

(mod-import mod dep)

pointer->tvm-ary clj

(pointer->tvm-ary ptr
                  device-type
                  device-id
                  datatype
                  shape
                  strides
                  byte-offset
                  &
                  [gc-root])

Take a pointer value and convert it into a dl-tensor datatype.

strides are optional, can be nil.

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.

set-current-thread-stream clj

(set-current-thread-stream stream)

sync-stream-with-host clj

(sync-stream-with-host stream)

sync-stream-with-stream clj

(sync-stream-with-stream stream)

tvm-array clj

(tvm-array jvm-ary)

tvm-array->jvm clj

(tvm-array->jvm tvm-ary)