MCPcopy Create free account
hub / github.com/apache/tvm-ffi / alloc_tensor

Function alloc_tensor

examples/kernel_library/tvm_ffi_utils.h:62–64  ·  view source on GitHub ↗

[alloc_tensor.begin] --- Tensor allocation helper ---

Source from the content-addressed store, hash-verified

60// [alloc_tensor.begin]
61// --- Tensor allocation helper ---
62inline ffi::Tensor alloc_tensor(const ffi::Shape& shape, DLDataType dtype, DLDevice device) {
63 return ffi::Tensor::FromEnvAlloc(TVMFFIEnvTensorAlloc, shape, dtype, device);
64}
65// [alloc_tensor.end]
66
67// [dtype_constants.begin]

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected