Interface ITensorAllocator

Interfaces for tensor allocator

Inherited Members
Namespace: Unity.Barracuda
Syntax
public interface ITensorAllocator : IDisposable

Methods

Alloc(TensorShape)

Declaration
Tensor Alloc(TensorShape shape)
Parameters
Type Name Description
TensorShape shape
Returns
Type Description
Tensor

Alloc(TensorShape, ITensorData)

Declaration
Tensor Alloc(TensorShape shape, ITensorData buffer)
Parameters
Type Name Description
TensorShape shape
ITensorData buffer
Returns
Type Description
Tensor

MoveToDevice(Tensor, ITensorData, ITensorData, Boolean)

Declaration
void MoveToDevice(Tensor x, ITensorData newBuffer, ITensorData oldBuffer, bool disposeDetachedBufferHint)
Parameters
Type Name Description
Tensor x
ITensorData newBuffer
ITensorData oldBuffer
Boolean disposeDetachedBufferHint

Release(Tensor, Boolean)

Declaration
void Release(Tensor x, bool calledFromTensorDispose)
Parameters
Type Name Description
Tensor x
Boolean calledFromTensorDispose

Reset(Boolean)

Declaration
void Reset(bool keepCachedMemory)
Parameters
Type Name Description
Boolean keepCachedMemory

WaiveOwnership(Tensor)

Declaration
void WaiveOwnership(Tensor x)
Parameters
Type Name Description
Tensor x