Liking cljdoc? Tell your friends :D
tvm-clj.api
Higher level API to build and compile tvm functions.
Higher level API to build and compile tvm functions.
- *varname-prefix*
- ->call-type
- ->operation
- ->stage
- abs
- add
- bind-arguments
- call
- call-intrin
- call-pure-intrin
- call-type-set
- call-types
- cast
- ceil
- commutative-reduce
- compute
- const
- create-schedule
- declare-buffer
- def-bin-intrin-op
- def-bin-op
- def-intrin-op
- def-op
- default-build-config
- div
- eq
- exp
- floor
- input-tensors
- int->lowered-function-type-map
- iteration-variable
- iteration-variable-type-set
- iteration-variable-types
- log
- lowered-function-type->int-map
- lowered-functions->module
- max
- min
- mod
- mul
- name->thread-axis-iterator
- node->str
- output-tensors
- placeholder
- popcount
- power
- range
- round
- safe-str
- schedule->lowered-function
- schedule->str
- schedule-cache-read
- schedule-cache-write
- schedules->fns
- select
- sigmoid
- sqrt
- stage-bind
- stage-bind-gpu
- stage-compute-at
- stage-cpu-injective
- stage-fuse
- stage-gpu-injective
- stage-inline
- stage-parallel
- stage-reorder
- stage-split-axis
- stage-tile
- stage-unroll
- stage-vectorize
- static-cast
- sub
- tanh
- target-info
- target-name->props
- target-name->thread-warp-size
- tget
- throw-nil
- trunc
- tvm-fn
- tvm-let
- variable
- when-not-error
tvm-clj.api-sugar
Operators and bindings to make the clojure code look and work like the python tvm bindings. This file is purely syntax sugar.
Operators and bindings to make the clojure code look and work like the python tvm bindings. This file is purely syntax sugar.
tvm-clj.bindings.definitions
- datatype->dl-type-code-map
- device-attribute-map
- device-type->device-type-int
- device-type->kwd-map
- device-type-int->device-type
- dl-dtype-map->datatype-map
- expression-set
- is-expression-node?
- keyword->tvm-datatype
- kwd->device-type-map
- node-type-name->keyword-map
- tvm-datatype->keyword
- tvm-datatype->keyword-map
- tvm-datatype->keyword-nothrow
tvm-clj.jna.base
- ->long-ptr
- arg-list->tvm-args
- call-function
- check-call
- datatype->dl-datatype
- datatype->dl-type-code
- device-id->int
- device-type->int
- dl-datatype->datatype
- dl-datatype->map
- fn-name
- get-last-error
- global-function
- global-function-names
- int-ptr
- keyword->tvm-datatype
- long-ptr
- make-tvm-jna-fn
- name->global-function
- ptr-ptr
- tvm-datatype->keyword
- tvm-datatype->keyword-nothrow
- tvm-value->jvm
- TVMFuncCall
- TVMFuncGetGlobal
- TVMFuncListGlobalNames
- TVMGetLastError
tvm-clj.tvm-jna
- ->node
- ->tvm
- ->tvm-value
- allocate-device-array
- base-ptr
- byte-offset
- call-function
- check-cpu-tensor
- copy-array-to-array!
- copy-from-array!
- copy-to-array!
- create-stream
- device-exists?
- device-id
- device-type
- device-type->int
- device-type-int->device-type
- g-fn
- get-module-function
- get-module-source
- get-node-type
- global-function
- global-node-function
- gn-fn
- is-expression-node?
- is-node-handle?
- mod-import
- pointer->tvm-ary
- set-current-thread-stream
- sync-stream-with-host
- sync-stream-with-stream
- tvm-array
- tvm-array->jvm
cljdoc is a website building & hosting documentation for Clojure/Script libraries
× close