saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-29 23:06:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Clean up ExecutionProvider in CSharp (#783)

Browse source
This commit is contained in:
Yufeng Li 2019-04-05 22:29:54 -07:00 committed by GitHub
parent fda1d0dce9
commit cea2a40bf1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 5 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
csharp/src/Microsoft.ML.OnnxRuntime/InferenceSession.cs
View file

@ -137,15 +137,15 @@ namespace Microsoft.ML.OnnxRuntime
var inputTensors = new IntPtr[inputs.Count];
var pinnedBufferHandles = new System.Buffers.MemoryHandle[inputs.Count];
int offset = 0;
int inputIndex = 0;
foreach (var input in inputs)
{
inputNames[offset] = input.Name;
inputNames[inputIndex] = input.Name;
// create Tensor from the input if feasible, else throw notsupported exception for now
input.ToNativeOnnxValue(out inputTensors[offset], out pinnedBufferHandles[offset]);
input.ToNativeOnnxValue(out inputTensors[inputIndex], out pinnedBufferHandles[inputIndex]);
offset++;
inputIndex++;
}
string[] outputNamesArray = outputNames.ToArray();

13
csharp/src/Microsoft.ML.OnnxRuntime/SessionOptions.cs
View file

@ -7,17 +7,6 @@ using System.Runtime.InteropServices;
namespace Microsoft.ML.OnnxRuntime
{
/// <summary>
/// Various providers of ONNX operators
/// </summary>
public enum ExecutionProvider
{
Cpu,
MklDnn,
Cuda
//TODO: add more providers gradually
};
/// <summary>
/// Holds the options for creating an InferenceSession
/// </summary>
@ -56,7 +45,7 @@ namespace Microsoft.ML.OnnxRuntime
/// </param>
public void EnableSequentialExecution()
{
NativeMethods.OrtEnableSequentialExecution(_nativePtr);
NativeMethods.OrtEnableSequentialExecution(_nativePtr);
}
/// <summary>

9
docs/CSharp_API.md
View file

@ -117,15 +117,6 @@ Enable Sequential Execution. By default, it is enabled.
DisableSequentialExecution();
Disable Sequential Execution and enable Parallel Execution.
AppendExecutionProvider(ExecutionProvider provider);
Appends execution provider to the session. For any operator in the graph the first execution provider that implements the operator will be user. ExecutionProvider is defined as the following enum.
enum ExecutionProvider
{
Cpu,
MklDnn
}
### NodeMetadata
Container of metadata for a model graph node, used for communicating the shape and type of the input and output nodes.