Class ModelExtensions
Extensions for Model
class
Inherited Members
Namespace: Unity.Barracuda
Syntax
public static class ModelExtensions
Methods
CreateWorker(Model, String[], String[], WorkerFactory.Device, Boolean)
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 = WorkerFactory.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 |
Boolean | verbose | will log scheduling of layers execution to the console (default == false) |
Returns
Type | Description |
---|---|
IWorker | Worker instance |
CreateWorker(Model, WorkerFactory.Device, Boolean)
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 = WorkerFactory.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 |
Boolean | verbose | will log scheduling of layers execution to the console |
Returns
Type | Description |
---|---|
IWorker | Worker instance |