Class ModelMetadataExtensions
Model metadata extensions
Namespace: Unity.Barracuda
Syntax
public static class ModelMetadataExtensions : object
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 |
|---|---|
| Int32 | 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 |
|---|---|
| Nullable<TensorShape> | Tensor shape |
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 |