Class ModelMetadataExtensions
Model metadata extensions
Inherited Members
Namespace: Unity.Barracuda
Assembly: Unity.Barracuda.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 |