Interface ITensorAllocator
An interface that provides methods for allocating tensors.
Inherited Members
Namespace: Unity.Sentis
Assembly: solution.dll
Syntax
public interface ITensorAllocator : IDisposableMethods
| Name | Description |
|---|---|
| Alloc(TensorShape, DataType, DeviceType, AllocScope) | Allocates a tensor of a given shape, data type on a given device type, and given scope. |
| Alloc(TensorShape, DataType, ITensorData, AllocScope) | Allocates a tensor of a given shape, data type, and a given scope from an existing |
| MoveToDevice(Tensor, ITensorData, ITensorData, bool) | Moves a tensor to a device. |
| PostLayerCleanup() | Allows ITensorAllocator to run cleanup operations such as clearing temporary buffers only used in the scope of the last layer executed. |
| Release(Tensor, bool) | Releases a tensor. |
| Reset(bool) | Resets the allocator. |
| WaiveOwnership(Tensor) | Waives ownership of a tensor. |