lib/onnxruntime/ffi.rb in onnxruntime-0.2.1 vs lib/onnxruntime/ffi.rb in onnxruntime-0.2.2
- old
+ new
@@ -1,11 +1,11 @@
module OnnxRuntime
module FFI
extend ::FFI::Library
begin
- ffi_lib OnnxRuntime.ffi_lib
+ ffi_lib Array(OnnxRuntime.ffi_lib)
rescue LoadError => e
raise e if ENV["ONNXRUNTIME_DEBUG"]
raise LoadError, "Could not find ONNX Runtime"
end
@@ -27,25 +27,25 @@
:DisableTelemetryEvents, callback(%i[pointer], :pointer),
:CreateSession, callback(%i[pointer string pointer pointer], :pointer),
:CreateSessionFromArray, callback(%i[pointer pointer size_t pointer pointer], :pointer),
:Run, callback(%i[pointer pointer pointer pointer size_t pointer size_t pointer], :pointer),
:CreateSessionOptions, callback(%i[pointer], :pointer),
- :SetOptimizedModelFilePath, callback(%i[], :pointer),
+ :SetOptimizedModelFilePath, callback(%i[pointer string], :pointer),
:CloneSessionOptions, callback(%i[], :pointer),
:SetSessionExecutionMode, callback(%i[], :pointer),
- :EnableProfiling, callback(%i[], :pointer),
- :DisableProfiling, callback(%i[], :pointer),
- :EnableMemPattern, callback(%i[], :pointer),
- :DisableMemPattern, callback(%i[], :pointer),
- :EnableCpuMemArena, callback(%i[], :pointer),
- :DisableCpuMemArena, callback(%i[], :pointer),
- :SetSessionLogId, callback(%i[], :pointer),
- :SetSessionLogVerbosityLevel, callback(%i[], :pointer),
- :SetSessionLogSeverityLevel, callback(%i[], :pointer),
- :SetSessionGraphOptimizationLevel, callback(%i[], :pointer),
- :SetIntraOpNumThreads, callback(%i[], :pointer),
- :SetInterOpNumThreads, callback(%i[], :pointer),
+ :EnableProfiling, callback(%i[pointer string], :pointer),
+ :DisableProfiling, callback(%i[pointer], :pointer),
+ :EnableMemPattern, callback(%i[pointer], :pointer),
+ :DisableMemPattern, callback(%i[pointer], :pointer),
+ :EnableCpuMemArena, callback(%i[pointer], :pointer),
+ :DisableCpuMemArena, callback(%i[pointer], :pointer),
+ :SetSessionLogId, callback(%i[pointer string], :pointer),
+ :SetSessionLogVerbosityLevel, callback(%i[pointer int], :pointer),
+ :SetSessionLogSeverityLevel, callback(%i[pointer int], :pointer),
+ :SetSessionGraphOptimizationLevel, callback(%i[pointer int], :pointer),
+ :SetIntraOpNumThreads, callback(%i[pointer int], :pointer),
+ :SetInterOpNumThreads, callback(%i[pointer int], :pointer),
:CreateCustomOpDomain, callback(%i[], :pointer),
:CustomOpDomain_Add, callback(%i[], :pointer),
:AddCustomOpDomain, callback(%i[], :pointer),
:RegisterCustomOpsLibrary, callback(%i[], :pointer),
:SessionGetInputCount, callback(%i[pointer pointer], :pointer),
@@ -55,18 +55,18 @@
:SessionGetOutputTypeInfo, callback(%i[pointer size_t pointer], :pointer),
:SessionGetOverridableInitializerTypeInfo, callback(%i[], :pointer),
:SessionGetInputName, callback(%i[pointer size_t pointer pointer], :pointer),
:SessionGetOutputName, callback(%i[pointer size_t pointer pointer], :pointer),
:SessionGetOverridableInitializerName, callback(%i[], :pointer),
- :CreateRunOptions, callback(%i[], :pointer),
- :RunOptionsSetRunLogVerbosityLevel, callback(%i[], :pointer),
- :RunOptionsSetRunLogSeverityLevel, callback(%i[], :pointer),
- :RunOptionsSetRunTag, callback(%i[], :pointer),
+ :CreateRunOptions, callback(%i[pointer], :pointer),
+ :RunOptionsSetRunLogVerbosityLevel, callback(%i[pointer int], :pointer),
+ :RunOptionsSetRunLogSeverityLevel, callback(%i[pointer int], :pointer),
+ :RunOptionsSetRunTag, callback(%i[pointer string], :pointer),
:RunOptionsGetRunLogVerbosityLevel, callback(%i[], :pointer),
:RunOptionsGetRunLogSeverityLevel, callback(%i[], :pointer),
:RunOptionsGetRunTag, callback(%i[], :pointer),
- :RunOptionsSetTerminate, callback(%i[], :pointer),
- :RunOptionsUnsetTerminate, callback(%i[], :pointer),
+ :RunOptionsSetTerminate, callback(%i[pointer], :pointer),
+ :RunOptionsUnsetTerminate, callback(%i[pointer], :pointer),
:CreateTensorAsOrtValue, callback(%i[pointer pointer size_t int pointer], :pointer),
:CreateTensorWithDataAsOrtValue, callback(%i[pointer pointer size_t pointer size_t int pointer], :pointer),
:IsTensor, callback(%i[], :pointer),
:GetTensorMutableData, callback(%i[pointer pointer], :pointer),
:FillStringTensor, callback(%i[pointer pointer size_t], :pointer),