(as-cpu-tensor data & {:keys [shape datatype]})
(assign! stream lhs rhs)
Broadcasting, marshalling assignment of rhs to lhs lhs = rhs
Broadcasting, marshalling assignment of rhs to lhs lhs = rhs
(assign-constant! stream tensor value)
(build-bind-map tensor-var-map)
The special cases in the bind map must relate to the name mangling of the function
The special cases in the bind map must relate to the name mangling of the function
(cpu-fallback-impl tm-fn stream & args)
(cpu-stream-fallback fn-name stream-item & args)
(explode-read-tensor tensor n-dims)
(get-fn-name stem dest & args)
(get-or-create-fn stream fn-name fn-create-fn)
(left-pad-ones shape-vec n-dims)
(n-dim-compute-op n-dims compute-fn & {:keys [name] :or {name "compute_op"}})
(n-dims->shape-stride-tuples n-dims arg-name)
(tensor-read placeholder index-vars shape-stride-tuples)
Perform the math required to get the absolute element offset from the shape/stride combined with the max shape variables
Perform the math required to get the absolute element offset from the shape/stride combined with the max shape variables
(tensor-read-arg->mangle-str tensor)
For arguments where we ourselves are interpreting the argument.
For arguments where we ourselves are interpreting the argument.
(tensor-read-dims->vars n-dims tensor arg-name)
(tensor-read-placeholder tensor arg-name)
(tensor-result-arg->mangle-str tensor)
(y-dim-tvm-fn n-dims detailed-fn)
cljdoc is a website building & hosting documentation for Clojure/Script libraries
× close