Interface IVars

Interfaces for variables

Namespace: Unity.Barracuda

Syntax

public interface IVars : IDisposable

Methods

DisposeAfterLayer(Layer)

Dispose storage that can be deleted after layer

Declaration

void DisposeAfterLayer(Layer forLayer)

Parameters

Type	Name	Description
Layer	forLayer	layer

GatherInputs(Layer)

Gather layer inputs

Declaration

Tensor[] GatherInputs(Layer forLayer)

Parameters

Type	Name	Description
Layer	forLayer	layer

Returns

Type	Description
Tensor[]	all input tensors

GetAllocator()

Get allocator

Declaration

ITensorAllocator GetAllocator()

Returns

Type	Description
ITensorAllocator	current ITensorAllocator

PeekConstants(String)

Peek constants

Declaration

Tensor[] PeekConstants(string layerName)

Parameters

Type	Name	Description
String	layerName	layer name

Returns

Type	Description
Tensor[]	Tensor array

PeekOutput(String)

Peek output

Declaration

Tensor PeekOutput(string name)

Parameters

Type	Name	Description
String	name	name

Returns

Type	Description
Tensor	Tensor

PrepareStorage(Layer)

Prepare storage for layer

Declaration

void PrepareStorage(Layer forLayer)

Parameters

Type	Name	Description
Layer	forLayer	layer

PrepareStorage(Model, IOps, IDictionary<String, TensorShape>, Boolean, DataType)

Prepare storage

Declaration

void PrepareStorage(Model model, IOps optionalOpsToPrepareTensors = null, IDictionary<string, TensorShape> optionalInputShapes = null, bool takeoverWeights = false, DataType dataType = DataType.Float)

Parameters

Type	Name	Description
Model	model	model
IOps	optionalOpsToPrepareTensors	IOps to prepare tensors
IDictionary<String, TensorShape>	optionalInputShapes	input shapes dictionary
Boolean	takeoverWeights	takeoverWeights flag
DataType	dataType	expect activation data type

SetInput(String, Tensor)

Set input

Declaration

void SetInput(string name, Tensor x)

Parameters

Type	Name	Description
String	name	name
Tensor	x	input

Store(Layer, Tensor)

Store result for layer

Declaration

void Store(Layer fromLayer, Tensor result)

Parameters

Type	Name	Description
Layer	fromLayer	layer
Tensor	result	Tensor to store