(->node item)
(->tvm item)
(->tvm-value item)
(allocate-device-array shape datatype device-type device-id)
(base-ptr item)
(byte-offset item)
(call-function tvm-fn & args)
(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)
(create-stream device-type device-id)
(device-exists? device-type device-id)
(device-id item)
(device-type item)
(device-type->int device-type)
(device-type-int->device-type device-type)
(get-module-function module fn-name & {:keys [query-imports?]})
(get-module-source module {:keys [format] :or {format ""}})
(get-node-type node-handle)
(global-function fn-name & args)
(is-expression-node? node)
(is-node-handle? item)
(mod-import mod dep)
(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.
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 stream)
(sync-stream-with-host stream)
(sync-stream-with-stream stream)
(tvm-array jvm-ary)
(tvm-array->jvm tvm-ary)
cljdoc is a website building & hosting documentation for Clojure/Script libraries
× close