Interface IWorker
Public interface for Workers. A worker is able to schedule models execution for a given backend.
Use WorkerFactory to instantiate a worker.
Namespace: Barracuda
Syntax
public interface IWorker : IDisposable
Methods
Execute()
Non-blocking API that schedules network execution in one go Remark: This API will only be non-blocking for GPU inference.
Declaration
void Execute()
Execute(Tensor)
Non-blocking API that schedules network execution in one go, using the provider tensor as input. Remark: This API will only be non-blocking for GPU inference. Useful when network have only one input as input name is not needed.
Declaration
void Execute(Tensor input)
Parameters
| Type | Name | Description |
|---|---|---|
| Tensor | input |
Execute(IDictionary<String, Tensor>)
Non-blocking API that schedules network execution in one go, using the provider tensor dictionary for inputs. Remark: This API will only be non-blocking for GPU inference.
Declaration
void Execute(IDictionary<string, Tensor> inputs)
Parameters
| Type | Name | Description |
|---|---|---|
| IDictionary<String, Tensor> | inputs |
ExecuteAsync()
Non-blocking API that schedules network execution one layer at the time. Remark: This API will only be non-blocking for GPU inference. Check GetAsyncProgress() for progress.
Declaration
IEnumerator ExecuteAsync()
Returns
| Type | Description |
|---|---|
| IEnumerator |
ExecuteAsync(Tensor)
Non-blocking API that schedules network execution one layer at the time, using the provider tensor as input. Remark: This API will only be non-blocking for GPU inference. Useful when network have only one input as input name is not needed. Check GetAsyncProgress() for progress.
Declaration
IEnumerator ExecuteAsync(Tensor input)
Parameters
| Type | Name | Description |
|---|---|---|
| Tensor | input |
Returns
| Type | Description |
|---|---|
| IEnumerator |
ExecuteAsync(IDictionary<String, Tensor>)
Non-blocking API that schedules network execution one layer at the time, using the provider tensor dictionary for inputs. Remark: This API will only be non-blocking for GPU inference. Check GetAsyncProgress() for progress.
Declaration
IEnumerator ExecuteAsync(IDictionary<string, Tensor> inputs)
Parameters
| Type | Name | Description |
|---|---|---|
| IDictionary<String, Tensor> | inputs |
Returns
| Type | Description |
|---|---|
| IEnumerator |
GetAsyncProgress()
Progress of the scheduling, 0.0 = 0%, 1.0 = 100%
Declaration
float GetAsyncProgress()
Returns
| Type | Description |
|---|---|
| Single |
PeekOutput()
Returns a reference to the first output tensor. This reference will be valid only until the next Execute() or Dispose() method is called on the worker.
Useful when network has only one output.
IMPORTANT: if you want tensor to outlive the worker, use CopyOutput() method or follow with TakeOwnership() call on the tensor.
Declaration
Tensor PeekOutput()
Returns
| Type | Description |
|---|---|
| Tensor |
PeekOutput(String)
Returns a reference to output tensor by name. This reference will be valid only until the next Execute() or Dispose() method is called on the worker.
IMPORTANT: if you want tensor to outlive the worker, use CopyOutput() method or follow with TakeOwnership() call on the tensor.
Declaration
Tensor PeekOutput(string name)
Parameters
| Type | Name | Description |
|---|---|---|
| String | name |
Returns
| Type | Description |
|---|---|
| Tensor |
PrepareForInput(IDictionary<String, TensorShape>)
Optional method to prepare network for particular input dimensions
Declaration
void PrepareForInput(IDictionary<string, TensorShape> inputShapes)
Parameters
| Type | Name | Description |
|---|---|---|
| IDictionary<String, TensorShape> | inputShapes |
SetInput(Tensor)
Specify single tensor value as the input for the network. Useful when network has only one input and caller does not need to know input's name.
Declaration
void SetInput(Tensor x)
Parameters
| Type | Name | Description |
|---|---|---|
| Tensor | x |
SetInput(String, Tensor)
Specify tensor value for the named input of the network.
Declaration
void SetInput(string name, Tensor x)
Parameters
| Type | Name | Description |
|---|---|---|
| String | name | |
| Tensor | x |
Summary()
Returns a string summary after execution.
Declaration
string Summary()
Returns
| Type | Description |
|---|---|
| String |
WaitForCompletion()
Wait for completion of part of the network that was scheduled via ExecuteAsync()
Declaration
void WaitForCompletion()