Class ModelExtensions
Extensions for Model class
Inherited Members
Namespace: Unity.Barracuda
Assembly: solution.dll
Syntax
public static class ModelExtensionsMethods
CreateWorker(Model, string[], string[], Device, bool)
Create a worker that will execute model using the best backend that is available for a given device type.
This is just a convenience function that internally calls ModelLoader.Load followed by ``WorkerFactory.CreateWorker`.
Declaration
public static IWorker CreateWorker(this Model model, string[] additionalOutputs, string[] trimOutputs, WorkerFactory.Device device = Device.Auto, bool verbose = false)Parameters
| Type | Name | Description |
|---|---|---|
| Model | model | the associated Model to execute |
| string[] | additionalOutputs | are the additional outputs to track but not directly specified by the model |
| string[] | trimOutputs | by specifying this list of outputs, all other non-specified outputs will be discarded |
| WorkerFactory.Device | device | the device type to run worker on. For example |
| bool | verbose | will log scheduling of layers execution to the console (default == false) |
Returns
| Type | Description |
|---|---|
| IWorker | Worker instance |
CreateWorker(Model, Device, bool)
Create a worker that will execute model using the best backend that is available for a given device type.
This is just a convenience function that internally calls ModelLoader.Load followed by ``WorkerFactory.CreateWorker`.
Declaration
public static IWorker CreateWorker(this Model model, WorkerFactory.Device device = Device.Auto, bool verbose = false)Parameters
| Type | Name | Description |
|---|---|---|
| Model | model | the associated Model to execute |
| WorkerFactory.Device | device | the preferred device for execution. For example |
| bool | verbose | will log scheduling of layers execution to the console |
Returns
| Type | Description |
|---|---|
| IWorker | Worker instance |