Class ModelMetadataExtensions
Model metadata extensions
Inherited Members
Namespace: Unity.Barracuda
Assembly: solution.dll
Syntax
public static class ModelMetadataExtensions
Methods
GetDownStreamLayersCount(Model, string)
Get count of layers that directly depend on specified input
Declaration
public static int GetDownStreamLayersCount(this Model model, string name)
Parameters
Type | Name | Description |
---|---|---|
Model | model | Model |
string | name | input name |
Returns
Type | Description |
---|---|
int | count of layers that directly depend on specified input |
GetShapeByName(Model, string)
Get model tensor shape by name
Declaration
public static TensorShape? GetShapeByName(this Model model, string name)
Parameters
Type | Name | Description |
---|---|---|
Model | model | Model |
string | name | Tensor name |
Returns
Type | Description |
---|---|
TensorShape? | Tensor shape |
Exceptions
Type | Condition |
---|---|
KeyNotFoundException |
GetTensorByName(Model, string)
Get model tensor by name
Declaration
public static Tensor GetTensorByName(this Model model, string name)
Parameters
Type | Name | Description |
---|---|---|
Model | model | Model |
string | name | Tensor name |
Returns
Type | Description |
---|---|
Tensor | Tensor |