Interface ITensorData

Interface for device dependent representation of Tensor data.

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

Properties

maxCapacity

Returns the maximum number of element this tensorData can contain.

Declaration
int maxCapacity { get; }
Property Value
Type Description
Int32

Methods

Download(TensorShape)

Returns an array filled with the values of a tensor. Depending on the implementation and underlying device this array might be a copy or direct reference to the tensor values. This is a blocking call, unless data from device was requested via ScheduleAsyncDownload beforehand and has already arrived. shape is the TensorShape (and thus length) of the data to copy.

Declaration
float[] Download(TensorShape shape)
Parameters
Type Name Description
TensorShape shape
Returns
Type Description
Single[]

Reserve(Int32)

Reserve uninitialized memory.

Declaration
void Reserve(int count)
Parameters
Type Name Description
Int32 count

ScheduleAsyncDownload(Int32)

Schedule an asynchronous download from device memory. count is the number of element to readback. returns false until data from device arrives to CPU and is ready for access.

Declaration
bool ScheduleAsyncDownload(int count)
Parameters
Type Name Description
Int32 count
Returns
Type Description
Boolean

SharedAccess(out Int32)

Returns an array filled with the values of multiple tensors that share the same tensorData on device. This function outputs offset from the beginning of the array to location of values for specific tensor. offset parameters is specified in float elements. Depending on the implementation and underlying device this array might be a copy or direct reference to tensor values, no conversion from on device memory layout will occur. This is a blocking call, unless data from device was requested via ScheduleAsyncDownload beforehand and has already arrived.

Declaration
float[] SharedAccess(out int offset)
Parameters
Type Name Description
Int32 offset
Returns
Type Description
Single[]

Upload(Single[], TensorShape, Int32)

Initialize with data. shape is the TensorShape (and thus length) of the data to copy. managedBufferStartIndex is the offset where to start the copy in the data

Declaration
void Upload(float[] data, TensorShape shape, int managedBufferStartIndex = 0)
Parameters
Type Name Description
Single[] data
TensorShape shape
Int32 managedBufferStartIndex

Extension Methods