Skip to main content
Log in

Mojo module

managed_tensor_slice

Implements the ManagedTensorSlice type - a view of a tensor that doesn't own the underlying data. This type is used to build custom graph operations.

Aliases

  • InputTensor = ManagedTensorSlice[IOSpec(), static_spec=?]:
  • MutableInputTensor = ManagedTensorSlice[IOSpec(), static_spec=?]:
  • OutputTensor = ManagedTensorSlice[IOSpec(), static_spec=?]:

Structs

  • DynamicTensor:
  • ManagedTensorSlice: A view of a tensor that does not own the underlying allocated pointer. When the object lifetime ends it does not free the underlying pointer. Conversely, if a ManagedTensorSlice is created, it will not extend the life of the underlying pointer.
  • VariadicTensors: A tuple-like container of tensors representing variadic arguments from the graph compiler.

Functions