2018-11-20 00:48:22 +00:00
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
using System ;
using System.Runtime.InteropServices ;
using System.Collections.Generic ;
using System.IO ;
using System.Linq ;
2020-08-10 20:33:49 +00:00
using Microsoft.ML.OnnxRuntime.Tensors ;
using System.Buffers ;
2018-11-20 00:48:22 +00:00
namespace Microsoft.ML.OnnxRuntime
{
/// <summary>
2020-11-20 22:03:55 +00:00
/// Represents an Inference Session on an ONNX Model.
/// This is a IDisposable class and it must be disposed of
/// using either a explicit call to Dispose() method or
/// a pattern of using() block. If this is a member of another
/// class that class must also become IDisposable and it must
/// dispose of InferfenceSession in its Dispose() method.
2018-11-20 00:48:22 +00:00
/// </summary>
2019-05-20 22:48:14 +00:00
public class InferenceSession : IDisposable
2018-11-20 00:48:22 +00:00
{
2020-11-20 22:03:55 +00:00
/// <summary>
/// A pointer to a underlying native instance of OrtSession
/// </summary>
2018-11-20 00:48:22 +00:00
protected IntPtr _nativeHandle ;
2020-11-20 22:03:55 +00:00
/// <summary>
/// Dictionaries that represent input/output/overridableInitializers metadata
/// </summary>
2019-10-04 23:38:00 +00:00
protected Dictionary < string , NodeMetadata > _inputMetadata , _outputMetadata , _overridableInitializerMetadata ;
2019-08-14 19:02:02 +00:00
private SessionOptions _builtInSessionOptions = null ;
private RunOptions _builtInRunOptions = null ;
2020-08-25 18:13:49 +00:00
private ModelMetadata _modelMetadata = null ;
2020-08-27 16:17:42 +00:00
private bool _disposed = false ;
2020-10-14 12:32:43 +00:00
private ulong _profilingStartTimeNs = 0 ;
2018-11-20 00:48:22 +00:00
#region Public API
2018-11-26 06:46:21 +00:00
/// <summary>
/// Constructs an InferenceSession from a model file
/// </summary>
/// <param name="modelPath"></param>
2018-11-20 00:48:22 +00:00
public InferenceSession ( string modelPath )
{
2019-08-14 19:02:02 +00:00
_builtInSessionOptions = new SessionOptions ( ) ; // need to be disposed
Init ( modelPath , _builtInSessionOptions ) ;
2018-11-20 00:48:22 +00:00
}
2021-05-15 03:44:42 +00:00
/// <summary>
/// Constructs an InferenceSession from a model file and it will use
/// the provided pre-packed weights container to store and share pre-packed buffers
/// of shared initializers across sessions if any.
/// </summary>
/// <param name="modelPath">Model path</param>
/// <param name="prepackedWeightsContainer">Instance of PrepackedWeightsContainer.
/// Lifetime of 'prepackedWeightsContainer' must be
/// managed by the user and it must outlive any sessions reliant on it</param>
public InferenceSession ( string modelPath , PrePackedWeightsContainer prepackedWeightsContainer )
{
_builtInSessionOptions = new SessionOptions ( ) ; // need to be disposed
Init ( modelPath , _builtInSessionOptions , prepackedWeightsContainer ) ;
}
2019-08-14 19:02:02 +00:00
2018-11-26 06:46:21 +00:00
/// <summary>
/// Constructs an InferenceSession from a model file, with some additional session options
/// </summary>
/// <param name="modelPath"></param>
/// <param name="options"></param>
2018-11-20 00:48:22 +00:00
public InferenceSession ( string modelPath , SessionOptions options )
{
2019-08-14 19:02:02 +00:00
Init ( modelPath , options ) ;
2018-11-20 00:48:22 +00:00
}
2021-05-15 03:44:42 +00:00
/// <summary>
/// Constructs an InferenceSession from a model file, with some additional session options
/// and it will use the provided pre-packed weights container to store and share pre-packed buffers
/// of shared initializers across sessions if any.
/// </summary>
/// <param name="modelPath">Model path</param>
/// <param name="options">Session options</param>
/// <param name="prepackedWeightsContainer">Instance of PrepackedWeightsContainer.
/// Lifetime of 'prepackedWeightsContainer' must be
/// managed by the user and it must outlive any sessions reliant on it</param>
public InferenceSession ( string modelPath , SessionOptions options ,
PrePackedWeightsContainer prepackedWeightsContainer )
{
Init ( modelPath , options , prepackedWeightsContainer ) ;
}
2019-09-24 18:59:04 +00:00
/// <summary>
/// Constructs an InferenceSession from a model data in byte array
/// </summary>
/// <param name="model"></param>
public InferenceSession ( byte [ ] model )
{
_builtInSessionOptions = new SessionOptions ( ) ; // need to be disposed
Init ( model , _builtInSessionOptions ) ;
}
2021-05-15 03:44:42 +00:00
/// <summary>
/// Constructs an InferenceSession from a model data (in byte array) and it will use
/// the provided pre-packed weights container to store and share pre-packed buffers
/// of shared initializers across sessions if any.
/// </summary>
/// <param name="model">Model as byte array</param>
/// <param name="prepackedWeightsContainer">Instance of PrepackedWeightsContainer.
/// Lifetime of 'prepackedWeightsContainer' must be
/// managed by the user and it must outlive any sessions reliant on it</param>
public InferenceSession ( byte [ ] model , PrePackedWeightsContainer prepackedWeightsContainer )
{
_builtInSessionOptions = new SessionOptions ( ) ; // need to be disposed
Init ( model , _builtInSessionOptions , prepackedWeightsContainer ) ;
}
2019-09-24 18:59:04 +00:00
/// <summary>
/// Constructs an InferenceSession from a model data in byte array, with some additional session options
/// </summary>
/// <param name="model"></param>
/// <param name="options"></param>
public InferenceSession ( byte [ ] model , SessionOptions options )
{
Init ( model , options ) ;
}
2019-08-14 19:02:02 +00:00
2021-05-15 03:44:42 +00:00
/// <summary>
/// Constructs an InferenceSession from a model data (in byte array) with some additional
/// session options and it will use the provided pre-packed weights container to store
/// and share pre-packed buffers of shared initializers across sessions if any.
/// </summary>
/// <param name="model">Model as byte array</param>
/// <param name="options">Session Options</param>
/// <param name="prepackedWeightsContainer">Instance of PrepackedWeightsContainer.
/// Lifetime of 'prepackedWeightsContainer' must be
/// managed by the user and it must outlive any sessions reliant on it</param>
public InferenceSession ( byte [ ] model , SessionOptions options ,
PrePackedWeightsContainer prepackedWeightsContainer )
{
Init ( model , options , prepackedWeightsContainer ) ;
}
2019-08-14 19:02:02 +00:00
/// <summary>
/// Meta data regarding the input nodes, keyed by input names
/// </summary>
2018-11-20 00:48:22 +00:00
public IReadOnlyDictionary < string , NodeMetadata > InputMetadata
{
get
{
2019-05-20 22:48:14 +00:00
return _inputMetadata ;
2018-11-20 00:48:22 +00:00
}
}
2019-08-14 19:02:02 +00:00
/// <summary>
/// Metadata regarding the output nodes, keyed by output names
/// </summary>
2018-11-20 00:48:22 +00:00
public IReadOnlyDictionary < string , NodeMetadata > OutputMetadata
{
get
{
2019-05-20 22:48:14 +00:00
return _outputMetadata ;
2018-11-20 00:48:22 +00:00
}
}
2019-10-04 23:38:00 +00:00
/// <summary>
/// Metadata regarding the overridable initializers, keyed by node names
/// </summary>
public IReadOnlyDictionary < string , NodeMetadata > OverridableInitializerMetadata
{
get
{
return _overridableInitializerMetadata ;
}
}
2019-08-14 19:02:02 +00:00
2018-11-26 06:46:21 +00:00
/// <summary>
/// Runs the loaded model for the given inputs, and fetches all the outputs.
/// </summary>
2020-04-08 18:57:40 +00:00
/// <param name="inputs">specify a collection of <see cref="NamedOnnxValue"/> that indicates the input values.</param>
2019-08-14 19:02:02 +00:00
/// <returns>Output Tensors in a Collection of NamedOnnxValue. User must dispose the output.</returns>
2019-01-29 05:40:19 +00:00
public IDisposableReadOnlyCollection < DisposableNamedOnnxValue > Run ( IReadOnlyCollection < NamedOnnxValue > inputs )
2018-11-23 04:56:43 +00:00
{
2018-11-26 06:46:21 +00:00
string [ ] outputNames = new string [ _outputMetadata . Count ] ;
_outputMetadata . Keys . CopyTo ( outputNames , 0 ) ;
return Run ( inputs , outputNames ) ;
2018-11-23 04:56:43 +00:00
}
2018-11-26 06:46:21 +00:00
/// <summary>
/// Runs the loaded model for the given inputs, and fetches the outputs specified in <paramref name="outputNames"/>.
/// </summary>
2020-04-08 18:57:40 +00:00
/// <param name="inputs">Specify a collection of <see cref="NamedOnnxValue"/> that indicates the input values.</param>
/// <param name="outputNames">Specify a collection of string that indicates the output names to fetch.</param>
2019-08-14 19:02:02 +00:00
/// <returns>Output Tensors in a Collection of NamedOnnxValue. User must dispose the output.</returns>
2019-01-29 05:40:19 +00:00
public IDisposableReadOnlyCollection < DisposableNamedOnnxValue > Run ( IReadOnlyCollection < NamedOnnxValue > inputs , IReadOnlyCollection < string > outputNames )
2018-11-20 00:48:22 +00:00
{
2020-04-08 18:57:40 +00:00
return Run ( inputs , outputNames , _builtInRunOptions ) ;
2018-11-20 00:48:22 +00:00
}
/// <summary>
2020-04-08 18:57:40 +00:00
/// Runs the loaded model for the given inputs, and fetches the specified outputs in <paramref name="outputNames"/>. Uses the given RunOptions for this run.
2018-11-20 00:48:22 +00:00
/// </summary>
2020-04-08 18:57:40 +00:00
/// <param name="inputs">Specify a collection of <see cref="NamedOnnxValue"/> that indicates the input values.</param>
/// <param name="outputNames">Specify a collection of string that indicates the output names to fetch.</param>
2018-11-20 00:48:22 +00:00
/// <param name="options"></param>
2019-08-14 19:02:02 +00:00
/// <returns>Output Tensors in a Collection of NamedOnnxValue. User must dispose the output.</returns>
public IDisposableReadOnlyCollection < DisposableNamedOnnxValue > Run ( IReadOnlyCollection < NamedOnnxValue > inputs , IReadOnlyCollection < string > outputNames , RunOptions options )
2018-11-20 00:48:22 +00:00
{
2020-08-10 20:33:49 +00:00
using ( var cleanupList = new DisposableList < IDisposable > ( ) )
2018-11-20 00:48:22 +00:00
{
2020-08-10 20:33:49 +00:00
var inputNamesArray = ConvertNamesToUtf8 ( inputs , v = > v . Name , cleanupList ) ;
var inputValuesArray = GetOrtValuesHandles ( inputs , cleanupList ) ;
var outputNamesArray = ConvertNamesToUtf8 ( outputNames , n = > n , cleanupList ) ;
2018-11-20 00:48:22 +00:00
2020-08-10 20:33:49 +00:00
var ortValues = RunImpl ( options , inputNamesArray , inputValuesArray , outputNamesArray , cleanupList ) ;
return CreateDisposableResult ( ortValues , outputNames ) ;
2018-11-20 00:48:22 +00:00
}
2020-04-08 18:57:40 +00:00
}
/// <summary>
/// Runs the loaded model for the given inputs, and fetches all the outputs.
/// </summary>
/// <param name="inputNames">Specify a collection of string that indicates the input names. Should match <paramref name="inputValues"/>.</param>
/// <param name="inputValues">Specify a collection of <see cref="FixedBufferOnnxValue"/> that indicates the input values.</param>
/// <returns>Output Tensors in a Collection of NamedOnnxValue. User must dispose the output.</returns>
public IDisposableReadOnlyCollection < DisposableNamedOnnxValue > Run (
IReadOnlyCollection < string > inputNames ,
IReadOnlyCollection < FixedBufferOnnxValue > inputValues )
{
string [ ] outputNames = new string [ _outputMetadata . Count ] ;
_outputMetadata . Keys . CopyTo ( outputNames , 0 ) ;
return Run ( inputNames , inputValues , outputNames , _builtInRunOptions ) ;
}
/// <summary>
/// Runs the loaded model for the given inputs, and fetches the outputs specified in <paramref name="outputNames"/>.
/// </summary>
/// <param name="inputNames">Specify a collection of string that indicates the input names. Should match <paramref name="inputValues"/>.</param>
/// <param name="inputValues">Specify a collection of <see cref="FixedBufferOnnxValue"/> that indicates the input values.</param>
/// <param name="outputNames">Specify a collection of string that indicates the output names to fetch.</param>
/// <returns>Output Tensors in a Collection of NamedOnnxValue. User must dispose the output.</returns>
public IDisposableReadOnlyCollection < DisposableNamedOnnxValue > Run (
IReadOnlyCollection < string > inputNames ,
IReadOnlyCollection < FixedBufferOnnxValue > inputValues ,
IReadOnlyCollection < string > outputNames )
{
return Run ( inputNames , inputValues , outputNames , _builtInRunOptions ) ;
}
/// <summary>
/// Runs the loaded model for the given inputs, and fetches the specified outputs in <paramref name="outputNames"/>. Uses the given RunOptions for this run.
/// </summary>
/// <param name="inputNames">Specify a collection of string that indicates the input names. Should match <paramref name="inputValues"/>.</param>
/// <param name="inputValues">Specify a collection of <see cref="FixedBufferOnnxValue"/> that indicates the input values.</param>
/// <param name="outputNames">Specify a collection of string that indicates the output names to fetch.</param>
/// <param name="options"></param>
/// <returns>Output Tensors in a Collection of NamedOnnxValue. User must dispose the output.</returns>
public IDisposableReadOnlyCollection < DisposableNamedOnnxValue > Run (
IReadOnlyCollection < string > inputNames ,
IReadOnlyCollection < FixedBufferOnnxValue > inputValues ,
IReadOnlyCollection < string > outputNames ,
RunOptions options )
{
if ( inputNames . Count ! = inputValues . Count )
{
throw new ArgumentException ( $"Length of {nameof(inputNames)} ({inputNames.Count}) must match that of {nameof(inputValues)} ({inputValues.Count})." ) ;
}
2020-08-10 20:33:49 +00:00
using ( var cleanupList = new DisposableList < IDisposable > ( ) )
2020-04-08 18:57:40 +00:00
{
2020-08-10 20:33:49 +00:00
var inputNamesArray = ConvertNamesToUtf8 ( inputNames , n = > n , cleanupList ) ;
IntPtr [ ] inputValuesArray = GetOrtValuesHandles ( inputValues , true ) ;
var outputNamesArray = ConvertNamesToUtf8 ( outputNames , n = > n , cleanupList ) ;
2020-04-08 18:57:40 +00:00
2020-08-10 20:33:49 +00:00
var ortValues = RunImpl ( options , inputNamesArray , inputValuesArray , outputNamesArray , cleanupList ) ;
return CreateDisposableResult ( ortValues , outputNames ) ;
2020-04-08 18:57:40 +00:00
}
}
/// <summary>
/// Runs the loaded model for the given inputs and outputs.
///
/// Outputs need to be created with correct type and dimension to accept the fetched data.
/// </summary>
/// <param name="inputNames">Specify a collection of string that indicates the input names. Should match <paramref name="inputValues"/>.</param>
/// <param name="inputValues">Specify a collection of <see cref="FixedBufferOnnxValue"/> that indicates the input values.</param>
/// <param name="outputNames">Specify a collection of string that indicates the output names. Should match <paramref name="outputValues"/>.</param>
/// <param name="outputValues">Specify a collection of <see cref="FixedBufferOnnxValue"/> that indicates the output values.</param>
public void Run (
IReadOnlyCollection < string > inputNames ,
IReadOnlyCollection < FixedBufferOnnxValue > inputValues ,
IReadOnlyCollection < string > outputNames ,
IReadOnlyCollection < FixedBufferOnnxValue > outputValues )
{
Run ( inputNames , inputValues , outputNames , outputValues , _builtInRunOptions ) ;
}
/// <summary>
/// Runs the loaded model for the given inputs and outputs. Uses the given RunOptions for this run.
///
/// Outputs need to be created with correct type and dimension to accept the fetched data.
/// </summary>
/// <param name="inputNames">Specify a collection of string that indicates the input names. Should match <paramref name="inputValues"/>.</param>
/// <param name="inputValues">Specify a collection of <see cref="FixedBufferOnnxValue"/> that indicates the input values.</param>
/// <param name="outputNames">Specify a collection of string that indicates the output names. Should match <paramref name="outputValues"/>.</param>
/// <param name="outputValues">Specify a collection of <see cref="FixedBufferOnnxValue"/> that indicates the output values.</param>
/// <param name="options"></param>
public void Run (
IReadOnlyCollection < string > inputNames ,
IReadOnlyCollection < FixedBufferOnnxValue > inputValues ,
IReadOnlyCollection < string > outputNames ,
IReadOnlyCollection < FixedBufferOnnxValue > outputValues ,
RunOptions options )
{
if ( inputNames . Count ! = inputValues . Count )
{
throw new ArgumentException ( $"Length of {nameof(inputNames)} ({inputNames.Count}) must match that of {nameof(inputValues)} ({inputValues.Count})." ) ;
}
if ( outputNames . Count ! = outputValues . Count )
{
throw new ArgumentException ( $"Length of {nameof(outputNames)} ({outputNames.Count}) must match that of {nameof(outputValues)} ({outputValues.Count})." ) ;
}
2020-08-10 20:33:49 +00:00
using ( var cleanupList = new DisposableList < IDisposable > ( ) )
2020-04-08 18:57:40 +00:00
{
2020-08-10 20:33:49 +00:00
// prepare inputs
var inputNamesArray = ConvertNamesToUtf8 ( inputNames , n = > n , cleanupList ) ;
IntPtr [ ] inputValuesArray = GetOrtValuesHandles ( inputValues , true ) ;
2020-06-16 18:06:11 +00:00
2020-08-10 20:33:49 +00:00
// prepare outputs
var outputNamesArray = ConvertNamesToUtf8 ( outputNames , n = > n , cleanupList ) ;
IntPtr [ ] outputValuesArray = GetOrtValuesHandles ( outputValues , false ) ;
2020-04-08 18:57:40 +00:00
2020-08-10 20:33:49 +00:00
NativeApiStatus . VerifySuccess ( NativeMethods . OrtRun (
_nativeHandle ,
options . Handle ,
inputNamesArray ,
inputValuesArray ,
( UIntPtr ) inputNames . Count ,
outputNamesArray ,
( UIntPtr ) outputNames . Count ,
outputValuesArray /* pointers to Pre-allocated OrtValue instances */
) ) ;
2020-04-08 18:57:40 +00:00
}
}
/// <summary>
/// Runs the loaded model for the given inputs and outputs.
///
/// Outputs need to be created with correct type and dimension to receive the fetched data.
/// </summary>
/// <param name="inputs">Specify a collection of <see cref="NamedOnnxValue"/> that indicates the input values.</param>
2020-11-20 22:03:55 +00:00
/// <param name="outputs">Specify a collection of <see cref="NamedOnnxValue"/> that indicates the output values.</param>
2020-04-08 18:57:40 +00:00
public void Run (
IReadOnlyCollection < NamedOnnxValue > inputs ,
IReadOnlyCollection < NamedOnnxValue > outputs )
{
Run ( inputs , outputs , _builtInRunOptions ) ;
}
/// <summary>
2020-08-10 20:33:49 +00:00
///
2020-04-08 18:57:40 +00:00
/// Runs the loaded model for the given inputs and outputs. Uses the given RunOptions for this run.
2020-08-10 20:33:49 +00:00
///
2020-04-08 18:57:40 +00:00
/// Outputs need to be created with correct type and dimension to receive the fetched data.
/// </summary>
/// <param name="inputs">Specify a collection of <see cref="NamedOnnxValue"/> that indicates the input values.</param>
2020-11-20 22:03:55 +00:00
/// <param name="outputs">Specify a collection of <see cref="NamedOnnxValue"/> that indicates the output values.</param>
2020-04-08 18:57:40 +00:00
/// <param name="options"></param>
public void Run (
IReadOnlyCollection < NamedOnnxValue > inputs ,
IReadOnlyCollection < NamedOnnxValue > outputs ,
RunOptions options )
{
2021-05-15 03:44:42 +00:00
using ( var cleanupList = new DisposableList < IDisposable > ( ) )
2020-04-08 18:57:40 +00:00
{
2020-08-10 20:33:49 +00:00
var inputNamesArray = ConvertNamesToUtf8 ( inputs , i = > i . Name , cleanupList ) ;
var inputValuesArray = GetOrtValuesHandles ( inputs , cleanupList ) ;
2020-04-08 18:57:40 +00:00
2020-08-10 20:33:49 +00:00
var outputNamesArray = ConvertNamesToUtf8 ( outputs , o = > o . Name , cleanupList ) ;
var outputValuesArray = GetOrtValuesHandles ( outputs , cleanupList ) ;
2020-04-08 18:57:40 +00:00
2020-08-10 20:33:49 +00:00
NativeApiStatus . VerifySuccess ( NativeMethods . OrtRun (
2020-04-08 18:57:40 +00:00
_nativeHandle ,
options . Handle ,
inputNamesArray ,
inputValuesArray ,
( UIntPtr ) inputs . Count ,
outputNamesArray ,
( UIntPtr ) outputs . Count ,
outputValuesArray /* pointers to Pre-allocated OrtValue instances */
2020-08-10 20:33:49 +00:00
) ) ;
2020-04-08 18:57:40 +00:00
}
}
/// <summary>
/// Runs the loaded model for the given inputs and outputs.
///
/// Outputs need to be created with correct type and dimension to receive the fetched data.
/// </summary>
/// <param name="inputs">Specify a collection of <see cref="NamedOnnxValue"/> that indicates the input values.</param>
/// <param name="outputNames">Specify a collection of string that indicates the output names. Should match <paramref name="outputValues"/>.</param>
/// <param name="outputValues">Specify a collection of <see cref="FixedBufferOnnxValue"/> that indicates the output values.</param>
public void Run (
IReadOnlyCollection < NamedOnnxValue > inputs ,
IReadOnlyCollection < string > outputNames ,
IReadOnlyCollection < FixedBufferOnnxValue > outputValues )
{
Run ( inputs , outputNames , outputValues , _builtInRunOptions ) ;
}
/// <summary>
/// Runs the loaded model for the given inputs and outputs. Uses the given RunOptions for this run.
///
/// Outputs need to be created with correct type and dimension to receive the fetched data.
/// </summary>
/// <param name="inputs">Specify a collection of <see cref="NamedOnnxValue"/> that indicates the input values.</param>
/// <param name="outputNames">Specify a collection of string that indicates the output names. Should match <paramref name="outputValues"/>.</param>
/// <param name="outputValues">Specify a collection of <see cref="FixedBufferOnnxValue"/> that indicates the output values.</param>
/// <param name="options"></param>
public void Run (
IReadOnlyCollection < NamedOnnxValue > inputs ,
IReadOnlyCollection < string > outputNames ,
IReadOnlyCollection < FixedBufferOnnxValue > outputValues ,
RunOptions options )
{
if ( outputNames . Count ! = outputValues . Count )
{
throw new ArgumentException ( $"Length of {nameof(outputNames)} ({outputNames.Count}) must match that of {nameof(outputValues)} ({outputValues.Count})." ) ;
}
2021-05-15 03:44:42 +00:00
using ( var cleanupList = new DisposableList < IDisposable > ( ) )
2020-04-08 18:57:40 +00:00
{
// prepare inputs
2020-08-10 20:33:49 +00:00
var inputNamesArray = ConvertNamesToUtf8 ( inputs , i = > i . Name , cleanupList ) ;
var inputValuesArray = GetOrtValuesHandles ( inputs , cleanupList ) ;
2020-04-08 18:57:40 +00:00
// prepare outputs
2020-08-10 20:33:49 +00:00
var outputNamesArray = ConvertNamesToUtf8 ( outputNames , n = > n , cleanupList ) ;
var outputValuesArray = GetOrtValuesHandles ( outputValues , false ) ;
2020-04-08 18:57:40 +00:00
2020-08-10 20:33:49 +00:00
NativeApiStatus . VerifySuccess ( NativeMethods . OrtRun (
2020-04-08 18:57:40 +00:00
_nativeHandle ,
options . Handle ,
inputNamesArray ,
inputValuesArray ,
( UIntPtr ) inputs . Count ,
outputNamesArray ,
( UIntPtr ) outputNames . Count ,
outputValuesArray /* pointers to Pre-allocated OrtValue instances */
2020-08-10 20:33:49 +00:00
) ) ;
2020-04-08 18:57:40 +00:00
}
}
/// <summary>
2020-08-10 20:33:49 +00:00
///
2020-04-08 18:57:40 +00:00
/// Runs the loaded model for the given inputs and outputs.
2020-08-10 20:33:49 +00:00
///
2020-04-08 18:57:40 +00:00
/// Outputs need to be created with correct type and dimension to receive the fetched data.
/// </summary>
/// <param name="inputNames">Specify a collection of string that indicates the input names. Should match <paramref name="inputValues"/>.</param>
/// <param name="inputValues">Specify a collection of <see cref="FixedBufferOnnxValue"/> that indicates the input values.</param>
2020-11-20 22:03:55 +00:00
/// <param name="outputs">Specify a collection of <see cref="NamedOnnxValue"/> that indicates the output values.</param>
2020-04-08 18:57:40 +00:00
public void Run (
IReadOnlyCollection < string > inputNames ,
IReadOnlyCollection < FixedBufferOnnxValue > inputValues ,
IReadOnlyCollection < NamedOnnxValue > outputs )
{
Run ( inputNames , inputValues , outputs , _builtInRunOptions ) ;
}
/// <summary>
2020-08-10 20:33:49 +00:00
///
2020-04-08 18:57:40 +00:00
/// Runs the loaded model for the given inputs and outputs. Uses the given RunOptions for this run.
2020-08-10 20:33:49 +00:00
///
2020-04-08 18:57:40 +00:00
/// Outputs need to be created with correct type and dimension to receive the fetched data.
/// </summary>
/// <param name="inputNames">Specify a collection of string that indicates the input names. Should match <paramref name="inputValues"/>.</param>
/// <param name="inputValues">Specify a collection of <see cref="FixedBufferOnnxValue"/> that indicates the input values.</param>
2020-11-20 22:03:55 +00:00
/// <param name="outputs">Specify a collection of <see cref="NamedOnnxValue"/> that indicates the output values.</param>
2020-04-08 18:57:40 +00:00
/// <param name="options"></param>
public void Run (
IReadOnlyCollection < string > inputNames ,
IReadOnlyCollection < FixedBufferOnnxValue > inputValues ,
IReadOnlyCollection < NamedOnnxValue > outputs ,
RunOptions options )
{
if ( inputNames . Count ! = inputValues . Count )
{
throw new ArgumentException ( $"Length of {nameof(inputNames)} ({inputNames.Count}) must match that of {nameof(inputValues)} ({inputValues.Count})." ) ;
2018-11-20 00:48:22 +00:00
}
2019-05-20 22:48:14 +00:00
2021-05-15 03:44:42 +00:00
using ( var cleanupList = new DisposableList < IDisposable > ( ) )
2020-04-08 18:57:40 +00:00
{
// prepare inputs
2020-08-10 20:33:49 +00:00
var inputNamesArray = ConvertNamesToUtf8 ( inputNames , n = > n , cleanupList ) ;
var inputValuesArray = GetOrtValuesHandles ( inputValues , true ) ;
2020-04-08 18:57:40 +00:00
// prepare outputs
2020-08-10 20:33:49 +00:00
var outputNamesArray = ConvertNamesToUtf8 ( outputs , o = > o . Name , cleanupList ) ;
var outputValuesArray = GetOrtValuesHandles ( outputs , cleanupList ) ;
2020-04-08 18:57:40 +00:00
2020-08-10 20:33:49 +00:00
NativeApiStatus . VerifySuccess ( NativeMethods . OrtRun (
2020-04-08 18:57:40 +00:00
_nativeHandle ,
options . Handle ,
inputNamesArray ,
inputValuesArray ,
( UIntPtr ) inputNames . Count ,
outputNamesArray ,
( UIntPtr ) outputs . Count ,
outputValuesArray /* pointers to Pre-allocated OrtValue instances */
2020-08-10 20:33:49 +00:00
) ) ;
}
}
/// <summary>
/// Create OrtIoBinding instance to bind pre-allocated buffers
/// to input/output
/// </summary>
2020-11-24 22:10:14 +00:00
/// <returns>A new instance of OrtIoBinding</returns>
2020-08-10 20:33:49 +00:00
public OrtIoBinding CreateIoBinding ( )
{
return new OrtIoBinding ( this ) ;
}
2020-04-08 18:57:40 +00:00
2020-08-10 20:33:49 +00:00
/// <summary>
/// This method runs inference on the OrtIoBinding instance
/// The method does not return anything. This is a lightweight version of
/// RunWithBindingAndNames(). When you bind pre-allocated buffers to the output values
/// you may not want to fetch the outputs since you already have access to them so you can spare
/// the expense of fetching them and pairing with names.
/// You can still fetch the outputs by calling OrtIOBinding.GetOutputValues()
/// </summary>
2020-11-24 22:10:14 +00:00
/// <param name="runOptions">runOptions</param>
/// <param name="ioBinding">ioBinding instance to use</param>
2020-08-10 20:33:49 +00:00
public void RunWithBinding ( RunOptions runOptions , OrtIoBinding ioBinding )
{
NativeApiStatus . VerifySuccess ( NativeMethods . OrtRunWithBinding ( Handle , runOptions . Handle , ioBinding . Handle ) ) ;
}
2020-04-08 18:57:40 +00:00
2020-08-10 20:33:49 +00:00
/// <summary>
/// This method return a collection of DisposableNamedOnnxValue as in other interfaces
/// Query names from OrtIoBinding object and pair then with the array of OrtValues returned
/// from OrtIoBinding.GetOutputValues()
///
/// </summary>
/// <param name="runOptions">RunOptions</param>
/// <param name="ioBinding">OrtIoBinding instance with bindings</param>
/// <param name="names">optional parameter. If you already know the names of the outputs you can save a native
/// call to retrieve output names. They will be paired with the returned OrtValues and combined into DisposbleNamedOnnxValues.
/// Otherwise, the method will retrieve output names from the OrtIoBinding instance.
/// It is an error if you supply a different number of names than the returned outputs</param>
2020-11-20 22:03:55 +00:00
/// <returns>A disposable collection of DisposableNamedOnnxValue that encapsulate output OrtValues</returns>
2020-08-10 20:33:49 +00:00
public IDisposableReadOnlyCollection < DisposableNamedOnnxValue > RunWithBindingAndNames ( RunOptions runOptions , OrtIoBinding ioBinding , string [ ] names = null )
{
NativeApiStatus . VerifySuccess ( NativeMethods . OrtRunWithBinding ( Handle , runOptions . Handle , ioBinding . Handle ) ) ;
using ( var ortValues = ioBinding . GetOutputValues ( ) )
{
string [ ] outputNames = names ;
if ( outputNames = = null )
2020-04-08 18:57:40 +00:00
{
2020-08-10 20:33:49 +00:00
outputNames = ioBinding . GetOutputNames ( ) ;
}
if ( outputNames . Length ! = ortValues . Count )
{
throw new OnnxRuntimeException ( ErrorCode . InvalidArgument ,
"Number of specified names: " + names . Length + " does not match the output number: " +
ortValues . Count ) ;
}
var result = new DisposableList < DisposableNamedOnnxValue > ( outputNames . Length ) ;
try
{
for ( int i = 0 ; i < outputNames . Length ; + + i )
2020-04-08 18:57:40 +00:00
{
2020-08-10 20:33:49 +00:00
var ortValue = ortValues . ElementAt ( i ) ;
2020-11-11 18:21:22 +00:00
result . Add ( DisposableNamedOnnxValue . CreateFromOrtValue ( outputNames [ i ] , ortValue ) ) ;
2020-04-08 18:57:40 +00:00
}
2021-05-15 03:44:42 +00:00
}
catch ( Exception e )
2020-08-10 20:33:49 +00:00
{
result . Dispose ( ) ;
throw e ;
2020-04-08 18:57:40 +00:00
}
2020-08-10 20:33:49 +00:00
return result ;
2020-04-08 18:57:40 +00:00
}
2018-11-20 00:48:22 +00:00
}
2020-05-19 02:47:52 +00:00
/// <summary>
2020-11-20 22:03:55 +00:00
/// Ends profiling for the session.
/// </summary>
/// <returns> Returns the profile file name.</returns>
2020-05-19 02:47:52 +00:00
public string EndProfiling ( )
{
IntPtr nameHandle = IntPtr . Zero ;
2020-08-10 20:33:49 +00:00
var allocator = OrtAllocator . DefaultInstance ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionEndProfiling ( _nativeHandle ,
allocator . Pointer ,
out nameHandle ) ) ;
2021-05-15 03:44:42 +00:00
using ( var allocation = new OrtMemoryAllocation ( allocator , nameHandle , 0 ) )
2020-08-10 20:33:49 +00:00
{
return NativeOnnxValueHelper . StringFromNativeUtf8 ( nameHandle ) ;
}
}
2020-05-19 02:47:52 +00:00
2020-08-10 20:33:49 +00:00
// Delegate for string extraction from an arbitrary input/output object
private delegate string NameExtractor < in TInput > ( TInput input ) ;
2020-05-19 02:47:52 +00:00
2020-08-10 20:33:49 +00:00
/// <summary>
/// Run helper
/// </summary>
/// <param name="names">names to convert to zero terminated utf8 and pin</param>
/// <param name="cleanupList">list to add pinned memory to for later disposal</param>
/// <returns></returns>
private IntPtr [ ] ConvertNamesToUtf8 < T > ( IReadOnlyCollection < T > inputs , NameExtractor < T > extractor ,
DisposableList < IDisposable > cleanupList )
{
var result = new IntPtr [ inputs . Count ] ;
for ( int i = 0 ; i < inputs . Count ; + + i )
2020-05-19 02:47:52 +00:00
{
2020-08-10 20:33:49 +00:00
var name = extractor ( inputs . ElementAt ( i ) ) ;
var utf8Name = NativeOnnxValueHelper . StringToZeroTerminatedUtf8 ( name ) ;
var pinnedHandle = new PinnedGCHandle ( GCHandle . Alloc ( utf8Name , GCHandleType . Pinned ) ) ;
result [ i ] = pinnedHandle . Pointer ;
cleanupList . Add ( pinnedHandle ) ;
2020-05-19 02:47:52 +00:00
}
2020-08-10 20:33:49 +00:00
return result ;
}
/// <summary>
/// This function obtains ortValues for NamedOnnxValue.
/// The problem with NamedOnnxValue is that it does not contain any Onnx (OrtValue)
/// so calling ToOrtValue creates a new instance of OrtValue that needs to be disposed.
/// The deriving object DisposableNamedValue actually contains and owns OrtValue and it returns
/// it.
/// </summary>
/// <param name="values"></param>
/// <param name="cleanupList"></param>
/// <returns></returns>
private IntPtr [ ] GetOrtValuesHandles ( IReadOnlyCollection < NamedOnnxValue > values , DisposableList < IDisposable > cleanupList )
{
IntPtr [ ] result = new IntPtr [ values . Count ] ;
for ( int inputIndex = 0 ; inputIndex < values . Count ; + + inputIndex )
{
var input = values . ElementAt ( inputIndex ) ;
MemoryHandle ? memHandle ;
var ortValue = input . ToOrtValue ( out memHandle ) ;
if ( memHandle . HasValue )
{
cleanupList . Add ( memHandle ) ;
}
cleanupList . Add ( ortValue ) ;
result [ inputIndex ] = ortValue . Handle ;
}
return result ;
}
private IntPtr [ ] GetOrtValuesHandles ( IReadOnlyCollection < FixedBufferOnnxValue > values , bool input )
{
var valuesArray = new IntPtr [ values . Count ] ;
for ( int index = 0 ; index < values . Count ; + + index )
2020-05-19 02:47:52 +00:00
{
2020-08-10 20:33:49 +00:00
var v = values . ElementAt ( index ) ;
if ( ! input & & v . ElementType = = Tensors . TensorElementType . String )
2020-05-19 02:47:52 +00:00
{
2020-08-10 20:33:49 +00:00
throw new NotSupportedException ( "Using string type FixedBufferOnnxValue in outputs is not supported." ) ;
2020-05-19 02:47:52 +00:00
}
2020-08-10 20:33:49 +00:00
valuesArray [ index ] = v . Value . Handle ;
2020-05-19 02:47:52 +00:00
}
2020-08-10 20:33:49 +00:00
return valuesArray ;
}
2020-05-19 02:47:52 +00:00
2020-08-10 20:33:49 +00:00
2021-05-15 03:44:42 +00:00
private DisposableList < OrtValue > RunImpl ( RunOptions options , IntPtr [ ] inputNames , IntPtr [ ] inputValues , IntPtr [ ] outputNames ,
DisposableList < IDisposable > cleanupList )
2020-08-10 20:33:49 +00:00
{
var ortValues = new DisposableList < OrtValue > ( outputNames . Length ) ;
cleanupList . Add ( ortValues ) ;
IntPtr [ ] outputValuesArray = new IntPtr [ outputNames . Length ] ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtRun (
_nativeHandle ,
options . Handle ,
inputNames ,
inputValues ,
( UIntPtr ) inputNames . Length ,
outputNames ,
( UIntPtr ) outputNames . Length ,
outputValuesArray /* Empty array is passed in to receive output OrtValue pointers */
) ) ;
foreach ( var v in outputValuesArray )
{
ortValues . Add ( new OrtValue ( v ) ) ;
}
return ortValues ;
}
IDisposableReadOnlyCollection < DisposableNamedOnnxValue > CreateDisposableResult ( List < OrtValue > ortValues ,
IReadOnlyCollection < string > outputNames )
{
var result = new DisposableList < DisposableNamedOnnxValue > ( outputNames . Count ) ;
try
{
for ( int i = 0 ; i < ortValues . Count ; i + + )
{
var ortValue = ortValues [ i ] ;
result . Add ( DisposableNamedOnnxValue . CreateFromOrtValue ( outputNames . ElementAt ( i ) , ortValue ) ) ;
}
}
catch ( OnnxRuntimeException e )
{
result . Dispose ( ) ;
throw e ;
}
return result ;
2020-05-19 02:47:52 +00:00
}
2020-04-08 18:57:40 +00:00
2020-11-20 22:03:55 +00:00
/// <summary>
/// This property queries model metadata, constructs
/// an instance of ModelMetadata and caches it
/// </summary>
/// <returns>Instance of ModelMetdata</returns>
2020-08-25 18:13:49 +00:00
public ModelMetadata ModelMetadata
2018-11-26 06:46:21 +00:00
{
get
{
2020-08-25 18:13:49 +00:00
if ( _modelMetadata ! = null )
{
return _modelMetadata ;
}
_modelMetadata = new ModelMetadata ( this ) ;
return _modelMetadata ;
2018-11-26 06:46:21 +00:00
}
}
2018-11-20 00:48:22 +00:00
2020-10-14 12:32:43 +00:00
/// <summary>
/// Return the nanoseconds of profiling's start time
/// On some platforms, this timer may not be as precise as nanoseconds
/// For instance, on Windows and MacOS, the precision will be ~100ns
/// </summary>
public ulong ProfilingStartTimeNs
{
2021-05-15 03:44:42 +00:00
get
{
return _profilingStartTimeNs ;
}
}
2020-10-14 12:32:43 +00:00
2018-11-20 00:48:22 +00:00
#endregion
#region private methods
2019-08-14 19:02:02 +00:00
2021-05-15 03:44:42 +00:00
private void Init ( string modelPath , SessionOptions options ,
PrePackedWeightsContainer prepackedWeightsContainer = null )
2019-08-14 19:02:02 +00:00
{
2020-10-21 17:32:13 +00:00
var envHandle = OrtEnv . Handle ;
2019-09-24 18:59:04 +00:00
var session = IntPtr . Zero ;
2021-05-15 03:44:42 +00:00
if ( prepackedWeightsContainer = = null )
{
NativeApiStatus . VerifySuccess ( NativeMethods . OrtCreateSession ( envHandle , NativeMethods . GetPlatformSerializedString ( modelPath ) ,
options . Handle , out session ) ) ;
}
else
{
NativeApiStatus . VerifySuccess ( NativeMethods . OrtCreateSessionWithPrepackedWeightsContainer (
envHandle , NativeMethods . GetPlatformSerializedString ( modelPath ) ,
options . Handle , prepackedWeightsContainer . Pointer , out session ) ) ;
}
2019-08-14 19:02:02 +00:00
2019-09-24 18:59:04 +00:00
InitWithSessionHandle ( session , options ) ;
}
2021-05-15 03:44:42 +00:00
private void Init ( byte [ ] modelData , SessionOptions options ,
PrePackedWeightsContainer prepackedWeightsContainer = null )
2019-09-24 18:59:04 +00:00
{
2020-10-21 17:32:13 +00:00
var envHandle = OrtEnv . Handle ;
2019-09-24 18:59:04 +00:00
var session = IntPtr . Zero ;
2021-05-15 03:44:42 +00:00
if ( prepackedWeightsContainer = = null )
{
NativeApiStatus . VerifySuccess ( NativeMethods . OrtCreateSessionFromArray ( envHandle , modelData , ( UIntPtr ) modelData . Length , options . Handle , out session ) ) ;
}
else
{
NativeApiStatus . VerifySuccess ( NativeMethods . OrtCreateSessionFromArrayWithPrepackedWeightsContainer (
envHandle , modelData , ( UIntPtr ) modelData . Length , options . Handle , prepackedWeightsContainer . Pointer ,
out session ) ) ;
}
2019-09-24 18:59:04 +00:00
InitWithSessionHandle ( session , options ) ;
}
/// <summary>
/// Initializes the session object with a native session handle
/// </summary>
2020-11-11 18:21:22 +00:00
/// <param name="session">Value of a native session object</param>
2019-09-24 18:59:04 +00:00
/// <param name="options">Session options</param>
private void InitWithSessionHandle ( IntPtr session , SessionOptions options )
{
_nativeHandle = session ;
2019-08-14 19:02:02 +00:00
try
{
// Initialize input/output metadata
_inputMetadata = new Dictionary < string , NodeMetadata > ( ) ;
_outputMetadata = new Dictionary < string , NodeMetadata > ( ) ;
2019-10-04 23:38:00 +00:00
_overridableInitializerMetadata = new Dictionary < string , NodeMetadata > ( ) ;
2019-08-14 19:02:02 +00:00
// get input count
UIntPtr inputCount = UIntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionGetInputCount ( _nativeHandle , out inputCount ) ) ;
2019-10-04 23:38:00 +00:00
// get all the input names and metadata
2019-08-14 19:02:02 +00:00
for ( ulong i = 0 ; i < ( ulong ) inputCount ; i + + )
{
var iname = GetInputName ( i ) ;
_inputMetadata [ iname ] = GetInputMetadata ( i ) ;
}
// get output count
UIntPtr outputCount = UIntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionGetOutputCount ( _nativeHandle , out outputCount ) ) ;
2019-10-04 23:38:00 +00:00
// get all the output names and metadata
2019-08-14 19:02:02 +00:00
for ( ulong i = 0 ; i < ( ulong ) outputCount ; i + + )
{
_outputMetadata [ GetOutputName ( i ) ] = GetOutputMetadata ( i ) ;
}
2019-10-04 23:38:00 +00:00
// get overridable initializer count
UIntPtr initilaizerCount = UIntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionGetOverridableInitializerCount ( _nativeHandle , out initilaizerCount ) ) ;
// get all the overridable initializer names and metadata
for ( ulong i = 0 ; i < ( ulong ) initilaizerCount ; i + + )
{
_overridableInitializerMetadata [ GetOverridableInitializerName ( i ) ] = GetOverridableInitializerMetadata ( i ) ;
}
2020-10-14 12:32:43 +00:00
// set profiling's start time
UIntPtr startTime = UIntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionGetProfilingStartTimeNs ( _nativeHandle ,
2021-05-15 03:44:42 +00:00
out startTime ) ) ;
_profilingStartTimeNs = ( ulong ) startTime ;
2019-08-14 19:02:02 +00:00
}
catch ( OnnxRuntimeException e )
{
if ( _nativeHandle ! = IntPtr . Zero )
{
NativeMethods . OrtReleaseSession ( _nativeHandle ) ;
_nativeHandle = IntPtr . Zero ;
}
throw e ;
}
2020-10-14 12:32:43 +00:00
_builtInRunOptions = new RunOptions ( ) ; // create a default built-in run option, and avoid creating a new one every run() call
2019-08-14 19:02:02 +00:00
}
2018-11-20 00:48:22 +00:00
private string GetOutputName ( ulong index )
{
2020-08-10 20:33:49 +00:00
var allocator = OrtAllocator . DefaultInstance ;
2018-11-20 00:48:22 +00:00
IntPtr nameHandle = IntPtr . Zero ;
string str = null ;
2020-08-10 20:33:49 +00:00
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionGetOutputName (
_nativeHandle ,
( UIntPtr ) index ,
allocator . Pointer ,
out nameHandle ) ) ;
2018-11-20 00:48:22 +00:00
2020-08-10 20:33:49 +00:00
using ( var ortAllocation = new OrtMemoryAllocation ( allocator , nameHandle , 0 ) )
2018-11-20 00:48:22 +00:00
{
2020-08-10 20:33:49 +00:00
str = NativeOnnxValueHelper . StringFromNativeUtf8 ( nameHandle ) ;
2018-11-20 00:48:22 +00:00
}
return str ;
}
private string GetInputName ( ulong index )
{
string str = null ;
2020-08-10 20:33:49 +00:00
var allocator = OrtAllocator . DefaultInstance ;
IntPtr nameHandle = IntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionGetInputName (
_nativeHandle ,
( UIntPtr ) index ,
allocator . Pointer ,
out nameHandle ) ) ;
2018-11-20 00:48:22 +00:00
2020-08-10 20:33:49 +00:00
using ( var ortAllocation = new OrtMemoryAllocation ( allocator , nameHandle , 0 ) )
2018-11-20 00:48:22 +00:00
{
2020-08-10 20:33:49 +00:00
str = NativeOnnxValueHelper . StringFromNativeUtf8 ( nameHandle ) ;
2018-11-20 00:48:22 +00:00
}
return str ;
}
2019-10-04 23:38:00 +00:00
private string GetOverridableInitializerName ( ulong index )
{
string str = null ;
2020-08-10 20:33:49 +00:00
var allocator = OrtAllocator . DefaultInstance ;
IntPtr nameHandle = IntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionGetOverridableInitializerName (
_nativeHandle ,
( UIntPtr ) index ,
allocator . Pointer ,
out nameHandle ) ) ;
2021-05-15 03:44:42 +00:00
using ( var ortAllocation = new OrtMemoryAllocation ( allocator , nameHandle , 0 ) )
2019-10-04 23:38:00 +00:00
{
2020-08-10 20:33:49 +00:00
str = NativeOnnxValueHelper . StringFromNativeUtf8 ( nameHandle ) ;
2019-10-04 23:38:00 +00:00
}
return str ;
}
2018-11-20 00:48:22 +00:00
2018-11-23 04:56:43 +00:00
private NodeMetadata GetInputMetadata ( ulong index )
{
IntPtr typeInfo = IntPtr . Zero ;
2020-08-10 20:33:49 +00:00
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionGetInputTypeInfo ( _nativeHandle , ( UIntPtr ) index , out typeInfo ) ) ;
2018-11-23 04:56:43 +00:00
try
{
return GetMetadataFromTypeInfo ( typeInfo ) ;
}
finally
{
2020-08-10 20:33:49 +00:00
NativeMethods . OrtReleaseTypeInfo ( typeInfo ) ;
2018-11-23 04:56:43 +00:00
}
}
2018-11-20 00:48:22 +00:00
2018-11-23 04:56:43 +00:00
private NodeMetadata GetOutputMetadata ( ulong index )
{
IntPtr typeInfo = IntPtr . Zero ;
2020-08-10 20:33:49 +00:00
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionGetOutputTypeInfo ( _nativeHandle , ( UIntPtr ) index , out typeInfo ) ) ;
2018-11-23 04:56:43 +00:00
try
{
return GetMetadataFromTypeInfo ( typeInfo ) ;
}
finally
{
2020-08-10 20:33:49 +00:00
NativeMethods . OrtReleaseTypeInfo ( typeInfo ) ;
2019-10-04 23:38:00 +00:00
}
}
private NodeMetadata GetOverridableInitializerMetadata ( ulong index )
{
IntPtr typeInfo = IntPtr . Zero ;
2020-08-10 20:33:49 +00:00
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionGetOverridableInitializerTypeInfo ( _nativeHandle , ( UIntPtr ) index , out typeInfo ) ) ;
2019-10-04 23:38:00 +00:00
try
{
return GetMetadataFromTypeInfo ( typeInfo ) ;
}
finally
{
2020-08-10 20:33:49 +00:00
NativeMethods . OrtReleaseTypeInfo ( typeInfo ) ;
2018-11-23 04:56:43 +00:00
}
}
2019-03-06 00:00:40 +00:00
internal static NodeMetadata GetMetadataFromTypeInfo ( IntPtr typeInfo )
2018-11-23 04:56:43 +00:00
{
2019-06-11 01:36:04 +00:00
OnnxValueType valueType ;
{
2020-08-10 20:33:49 +00:00
IntPtr valType ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtGetOnnxTypeFromTypeInfo ( typeInfo , out valType ) ) ;
valueType = ( OnnxValueType ) valType ;
2019-06-11 01:36:04 +00:00
}
2019-03-12 17:11:14 +00:00
if ( valueType ! = OnnxValueType . ONNX_TYPE_TENSOR & & valueType ! = OnnxValueType . ONNX_TYPE_SPARSETENSOR )
{
2020-03-24 01:36:12 +00:00
return new NodeMetadata ( valueType , new int [ ] { } , new string [ ] { } , typeof ( NamedOnnxValue ) ) ;
2019-03-12 17:11:14 +00:00
}
2020-08-10 20:33:49 +00:00
// This should not be released
2019-06-11 01:36:04 +00:00
IntPtr tensorInfo ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtCastTypeInfoToTensorInfo ( typeInfo , out tensorInfo ) ) ; //(IntPtr)(int)(uint)
2019-03-06 00:00:40 +00:00
// Convert the newly introduced OrtTypeInfo* to the older OrtTypeAndShapeInfo*
if ( tensorInfo = = IntPtr . Zero )
return null ;
2018-11-23 04:56:43 +00:00
2019-06-11 01:36:04 +00:00
TensorElementType type ;
{
2020-08-10 20:33:49 +00:00
IntPtr el_type ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtGetTensorElementType ( tensorInfo , out el_type ) ) ;
type = ( TensorElementType ) el_type ;
2019-06-11 01:36:04 +00:00
}
2018-11-23 04:56:43 +00:00
Type dotnetType = null ;
int width = 0 ;
2019-05-20 22:48:14 +00:00
TensorElementTypeConverter . GetTypeAndWidth ( type , out dotnetType , out width ) ;
2019-06-11 01:36:04 +00:00
UIntPtr numDimensions ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtGetDimensionsCount ( tensorInfo , out numDimensions ) ) ;
2019-10-12 22:46:28 +00:00
2018-11-23 04:56:43 +00:00
long [ ] dimensions = new long [ ( int ) numDimensions ] ;
2019-10-12 22:46:28 +00:00
NativeApiStatus . VerifySuccess ( NativeMethods . OrtGetDimensions ( tensorInfo , dimensions , numDimensions ) ) ;
2018-11-23 04:56:43 +00:00
int [ ] intDimensions = new int [ ( int ) numDimensions ] ;
2019-05-20 22:48:14 +00:00
for ( var i = 0 ; i < ( long ) numDimensions ; i + + )
2018-11-23 04:56:43 +00:00
{
intDimensions [ i ] = ( int ) dimensions [ i ] ;
}
2019-10-12 22:46:28 +00:00
IntPtr [ ] dimensionNamePtrs = new IntPtr [ ( int ) numDimensions ] ;
NativeApiStatus . VerifySuccess (
NativeMethods . OrtGetSymbolicDimensions ( tensorInfo , dimensionNamePtrs , numDimensions ) ) ;
string [ ] symbolicDimensions = new string [ ( int ) numDimensions ] ;
for ( var i = 0 ; i < ( int ) numDimensions ; i + + )
{
2020-08-10 20:33:49 +00:00
symbolicDimensions [ i ] = NativeOnnxValueHelper . StringFromNativeUtf8 ( dimensionNamePtrs [ i ] ) ;
2019-10-12 22:46:28 +00:00
}
2020-03-24 01:36:12 +00:00
2019-10-12 22:46:28 +00:00
return new NodeMetadata ( valueType , intDimensions , symbolicDimensions , dotnetType ) ;
2018-11-23 04:56:43 +00:00
}
2020-08-10 20:33:49 +00:00
/// <summary>
/// Other classes access
/// </summary>
internal IntPtr Handle
2018-11-20 00:48:22 +00:00
{
2020-08-10 20:33:49 +00:00
get
{
return _nativeHandle ;
}
2018-11-20 00:48:22 +00:00
}
2020-08-10 20:33:49 +00:00
#endregion
2020-08-27 16:17:42 +00:00
#region IDisposable
/// <summary>
/// Finalizer. to cleanup session in case it runs
/// and the user forgets to Dispose() of the session
/// </summary>
~ InferenceSession ( )
{
Dispose ( false ) ;
}
2020-08-10 20:33:49 +00:00
2020-11-20 22:03:55 +00:00
/// <summary>
/// IDisposable implementation
/// </summary>
2018-11-20 00:48:22 +00:00
public void Dispose ( )
{
Dispose ( true ) ;
2020-08-27 16:17:42 +00:00
GC . SuppressFinalize ( this ) ;
2018-11-20 00:48:22 +00:00
}
2020-11-20 22:03:55 +00:00
/// <summary>
/// IDisposable implementation
/// </summary>
/// <param name="disposing">true if invoked from Dispose() method</param>
2018-11-20 00:48:22 +00:00
protected virtual void Dispose ( bool disposing )
{
2021-05-15 03:44:42 +00:00
if ( _disposed )
2020-08-27 16:17:42 +00:00
{
return ;
}
2018-11-20 00:48:22 +00:00
if ( disposing )
{
// cleanup managed resources
2019-08-14 19:02:02 +00:00
if ( _builtInSessionOptions ! = null )
{
_builtInSessionOptions . Dispose ( ) ;
2020-08-27 16:17:42 +00:00
_builtInSessionOptions = null ;
2019-08-14 19:02:02 +00:00
}
if ( _builtInRunOptions ! = null )
{
_builtInRunOptions . Dispose ( ) ;
2020-08-27 16:17:42 +00:00
_builtInRunOptions = null ;
2019-08-14 19:02:02 +00:00
}
2018-11-20 00:48:22 +00:00
}
// cleanup unmanaged resources
if ( _nativeHandle ! = IntPtr . Zero )
{
2018-12-18 19:39:46 +00:00
NativeMethods . OrtReleaseSession ( _nativeHandle ) ;
2020-08-27 16:17:42 +00:00
_nativeHandle = IntPtr . Zero ;
2018-11-20 00:48:22 +00:00
}
2020-08-27 16:17:42 +00:00
_disposed = true ;
2018-11-20 00:48:22 +00:00
}
#endregion
}
2018-11-23 04:56:43 +00:00
/// <summary>
/// Resembles type and shape information of session-graph nodes, used for communicating the shape/type of input/output nodes
/// </summary>
public class NodeMetadata
2018-11-20 00:48:22 +00:00
{
2019-10-12 22:46:28 +00:00
internal NodeMetadata ( OnnxValueType onnxValueType , int [ ] dimensions , string [ ] symbolicDimensions , Type type )
2018-11-20 00:48:22 +00:00
{
2020-11-20 22:03:55 +00:00
OnnxValueType = onnxValueType ;
Dimensions = dimensions ;
SymbolicDimensions = symbolicDimensions ;
ElementType = type ;
2018-11-23 04:56:43 +00:00
}
2020-11-20 22:03:55 +00:00
/// <summary>
/// Type value of the node
/// </summary>
/// <value>A value of OnnxValueType enum</value>
public OnnxValueType OnnxValueType { get ; }
2019-03-12 17:11:14 +00:00
2020-11-20 22:03:55 +00:00
/// <summary>
/// Shape
/// </summary>
/// <value>Array of dimensions</value>
public int [ ] Dimensions { get ; }
2019-10-12 22:46:28 +00:00
2020-11-20 22:03:55 +00:00
/// <summary>
/// Symbolic dimensions
/// </summary>
/// <value>Array of symbolic dimensions if present.</value>
public string [ ] SymbolicDimensions { get ; }
2019-10-12 22:46:28 +00:00
2020-11-20 22:03:55 +00:00
/// <summary>
/// .NET type that corresponds to this Node.
/// </summary>
/// <value>System.Type</value>
public System . Type ElementType { get ; }
2018-11-26 06:46:21 +00:00
2020-11-20 22:03:55 +00:00
/// <summary>
/// Whether it is a Tensor
/// </summary>
/// <value>currently always returns true</value>
2018-11-26 06:46:21 +00:00
public bool IsTensor
{
get
{
return true ; // currently only Tensor nodes are supported
}
}
2018-11-20 00:48:22 +00:00
}
2020-11-20 22:03:55 +00:00
/// <summary>
/// A class that queries and caches model metadata and exposes
/// it as properties
/// </summary>
2020-08-25 18:13:49 +00:00
public class ModelMetadata
2018-11-20 00:48:22 +00:00
{
2020-08-25 18:13:49 +00:00
private string _producerName ;
private string _graphName ;
private string _domain ;
private string _description ;
2021-01-06 06:18:03 +00:00
private string _graphDescription ;
2020-08-25 18:13:49 +00:00
private long _version ;
private Dictionary < string , string > _customMetadataMap = new Dictionary < string , string > ( ) ;
internal ModelMetadata ( InferenceSession session )
{
IntPtr modelMetadataHandle = IntPtr . Zero ;
var allocator = OrtAllocator . DefaultInstance ;
// Get the native ModelMetadata instance associated with the InferenceSession
NativeApiStatus . VerifySuccess ( NativeMethods . OrtSessionGetModelMetadata ( session . Handle , out modelMetadataHandle ) ) ;
try
{
// Process producer name
IntPtr producerNameHandle = IntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtModelMetadataGetProducerName ( modelMetadataHandle , allocator . Pointer , out producerNameHandle ) ) ;
using ( var ortAllocation = new OrtMemoryAllocation ( allocator , producerNameHandle , 0 ) )
{
_producerName = NativeOnnxValueHelper . StringFromNativeUtf8 ( producerNameHandle ) ;
}
// Process graph name
IntPtr graphNameHandle = IntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtModelMetadataGetGraphName ( modelMetadataHandle , allocator . Pointer , out graphNameHandle ) ) ;
using ( var ortAllocation = new OrtMemoryAllocation ( allocator , graphNameHandle , 0 ) )
{
_graphName = NativeOnnxValueHelper . StringFromNativeUtf8 ( graphNameHandle ) ;
}
// Process domain
IntPtr domainHandle = IntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtModelMetadataGetDomain ( modelMetadataHandle , allocator . Pointer , out domainHandle ) ) ;
using ( var ortAllocation = new OrtMemoryAllocation ( allocator , domainHandle , 0 ) )
{
_domain = NativeOnnxValueHelper . StringFromNativeUtf8 ( domainHandle ) ;
}
// Process description
IntPtr descriptionHandle = IntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtModelMetadataGetDescription ( modelMetadataHandle , allocator . Pointer , out descriptionHandle ) ) ;
using ( var ortAllocation = new OrtMemoryAllocation ( allocator , descriptionHandle , 0 ) )
{
_description = NativeOnnxValueHelper . StringFromNativeUtf8 ( descriptionHandle ) ;
}
2021-01-06 06:18:03 +00:00
// Process graph description
IntPtr graphDescriptionHandle = IntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtModelMetadataGetGraphDescription ( modelMetadataHandle , allocator . Pointer , out graphDescriptionHandle ) ) ;
using ( var ortAllocation = new OrtMemoryAllocation ( allocator , graphDescriptionHandle , 0 ) )
{
_graphDescription = NativeOnnxValueHelper . StringFromNativeUtf8 ( graphDescriptionHandle ) ;
}
2020-08-25 18:13:49 +00:00
// Process version
NativeApiStatus . VerifySuccess ( NativeMethods . OrtModelMetadataGetVersion ( modelMetadataHandle , out _version ) ) ;
// Process CustomMetadata Map
IntPtr customMetadataMapKeysHandle = IntPtr . Zero ;
long numKeys ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtModelMetadataGetCustomMetadataMapKeys ( modelMetadataHandle , allocator . Pointer , out customMetadataMapKeysHandle , out numKeys ) ) ;
// We have received an array of null terminated C strings which are the keys that we can use to lookup the custom metadata map
// The OrtAllocator will finally free the customMetadataMapKeysHandle
using ( var ortAllocationKeysArray = new OrtMemoryAllocation ( allocator , customMetadataMapKeysHandle , 0 ) )
using ( var ortAllocationKeys = new DisposableList < OrtMemoryAllocation > ( ( int ) numKeys ) )
{
// Put all the handles to each key in the DisposableList to be disposed off in an exception-safe manner
for ( int i = 0 ; i < ( int ) numKeys ; + + i )
{
ortAllocationKeys . Add ( new OrtMemoryAllocation ( allocator , Marshal . ReadIntPtr ( customMetadataMapKeysHandle , IntPtr . Size * i ) , 0 ) ) ;
}
// Process each key via the stored key handles
2021-05-15 03:44:42 +00:00
foreach ( var allocation in ortAllocationKeys )
2020-08-25 18:13:49 +00:00
{
IntPtr keyHandle = allocation . Pointer ;
IntPtr valueHandle = IntPtr . Zero ;
NativeApiStatus . VerifySuccess ( NativeMethods . OrtModelMetadataLookupCustomMetadataMap ( modelMetadataHandle , allocator . Pointer , keyHandle , out valueHandle ) ) ;
using ( var ortAllocationValue = new OrtMemoryAllocation ( allocator , valueHandle , 0 ) )
{
var key = NativeOnnxValueHelper . StringFromNativeUtf8 ( keyHandle ) ;
var value = NativeOnnxValueHelper . StringFromNativeUtf8 ( valueHandle ) ;
// Put the key/value pair into the dictionary
_customMetadataMap [ key ] = value ;
}
}
}
}
finally
{
// Free ModelMetadata handle
2021-05-15 03:44:42 +00:00
NativeMethods . OrtReleaseModelMetadata ( modelMetadataHandle ) ;
2020-08-25 18:13:49 +00:00
2021-05-15 03:44:42 +00:00
}
2020-08-25 18:13:49 +00:00
}
2020-11-20 22:03:55 +00:00
/// <summary>
/// Producer name string
/// </summary>
/// <value>producer name string</value>
2020-08-25 18:13:49 +00:00
public string ProducerName
{
get
{
return _producerName ;
}
}
2020-11-20 22:03:55 +00:00
/// <summary>
/// Graph name for this model
/// </summary>
/// <value>graph name string</value>
2020-08-25 18:13:49 +00:00
public string GraphName
{
get
{
return _graphName ;
}
}
2020-11-20 22:03:55 +00:00
/// <summary>
/// Domain for this model
/// </summary>
/// <value>domain name string</value>
2020-08-25 18:13:49 +00:00
public string Domain
{
get
{
return _domain ;
}
}
2020-11-20 22:03:55 +00:00
/// <summary>
/// Unstructured model description
/// </summary>
/// <value>description string</value>
2020-08-25 18:13:49 +00:00
public string Description
{
get
{
return _description ;
}
}
2021-01-06 06:18:03 +00:00
/// <summary>
/// Unstructured graph description
/// </summary>
/// <value>description string</value>
public string GraphDescription
{
get
{
return _graphDescription ;
}
}
2020-11-20 22:03:55 +00:00
/// <summary>
/// Version number
/// </summary>
/// <value>long version integer</value>
2020-08-25 18:13:49 +00:00
public long Version
{
get
{
return _version ;
}
}
2020-11-20 22:03:55 +00:00
/// <summary>
/// Custom metadata key/value pairs
/// </summary>
/// <value>An instance of a Dictionary<string,string></value>
2020-08-25 18:13:49 +00:00
public Dictionary < string , string > CustomMetadataMap
{
get
{
return _customMetadataMap ;
}
}
2018-11-20 00:48:22 +00:00
}
2018-11-23 04:56:43 +00:00
2018-11-20 00:48:22 +00:00
}