tvm_ffi::collections

Module tensor

Source

Structs§

  • CPUNDAlloc
    Example CPU NDAllocator This allocator allocates data on the CPU
  • Tensor
    ABI stable owned Shape for ffi
  • TensorObj
  • TensorView
    A non-owning view of a tensor: a pointer to a DLTensor.

Traits§

  • DLTensorExt
    DLTensorExt trait This trait provides methods to get the number of elements and the item size of a DLTensor
  • NDAllocator
    Trait for n-dimensional array allocators