Interface IVars
Interfaces for variables
Inherited Members
Namespace: Unity.Barracuda
Syntax
public interface IVars : IDisposableMethods
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 |
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 |
|
| 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 |