onnxruntime 0.3.0 → 0.3.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/CHANGELOG.md +5 -0
- data/lib/onnxruntime/ffi.rb +7 -1
- data/lib/onnxruntime/inference_session.rb +13 -3
- data/lib/onnxruntime/version.rb +1 -1
- data/vendor/libonnxruntime.dylib +0 -0
- data/vendor/libonnxruntime.so +0 -0
- data/vendor/onnxruntime.dll +0 -0
- metadata +2 -2
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: 2d4b9ddc710644eab07227fe005c92ad039643e989236e5f112a83df2266fc55
|
4
|
+
data.tar.gz: d15c9ac4ae98130499d1411d3be2e7b822df8248550c0ad44c2e7a02d88f95d8
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 83a2e3699506460b8b280376ec8935a680e7e2cffe6bf7dd6766a85eea84bf06273e13889b6a2961c9df3c277be6ea7d8afbb67a4d4be62d50b266d7ccd8dc55
|
7
|
+
data.tar.gz: 5086c2c04f76ec67224e7dbc9ccc36366b394840296536232b29d92009c7a4ef637fb4f39daec8abb7a5e3fdcca591675cfaf3e4a8e31ec939f9aa1e2ae6f112
|
data/CHANGELOG.md
CHANGED
data/lib/onnxruntime/ffi.rb
CHANGED
@@ -136,7 +136,13 @@ module OnnxRuntime
|
|
136
136
|
:ModelMetadataGetDescription, callback(%i[pointer pointer pointer], :pointer),
|
137
137
|
:ModelMetadataLookupCustomMetadataMap, callback(%i[pointer pointer pointer pointer], :pointer),
|
138
138
|
:ModelMetadataGetVersion, callback(%i[pointer pointer], :pointer),
|
139
|
-
:ReleaseModelMetadata, callback(%i[pointer], :void)
|
139
|
+
:ReleaseModelMetadata, callback(%i[pointer], :void),
|
140
|
+
:CreateEnvWithGlobalThreadPools, callback(%i[], :pointer),
|
141
|
+
:DisablePerSessionThreads, callback(%i[], :pointer),
|
142
|
+
:CreateThreadingOptions, callback(%i[], :pointer),
|
143
|
+
:ReleaseThreadingOptions, callback(%i[], :pointer),
|
144
|
+
:ModelMetadataGetCustomMetadataMapKeys, callback(%i[pointer pointer pointer pointer], :pointer),
|
145
|
+
:AddFreeDimensionOverrideByName, callback(%i[], :pointer)
|
140
146
|
end
|
141
147
|
|
142
148
|
class ApiBase < ::FFI::Struct
|
@@ -109,6 +109,8 @@ module OnnxRuntime
|
|
109
109
|
end
|
110
110
|
|
111
111
|
def modelmeta
|
112
|
+
keys = ::FFI::MemoryPointer.new(:pointer)
|
113
|
+
num_keys = ::FFI::MemoryPointer.new(:int64_t)
|
112
114
|
description = ::FFI::MemoryPointer.new(:string)
|
113
115
|
domain = ::FFI::MemoryPointer.new(:string)
|
114
116
|
graph_name = ::FFI::MemoryPointer.new(:string)
|
@@ -117,6 +119,16 @@ module OnnxRuntime
|
|
117
119
|
|
118
120
|
metadata = ::FFI::MemoryPointer.new(:pointer)
|
119
121
|
check_status api[:SessionGetModelMetadata].call(read_pointer, metadata)
|
122
|
+
|
123
|
+
custom_metadata_map = {}
|
124
|
+
check_status = api[:ModelMetadataGetCustomMetadataMapKeys].call(metadata.read_pointer, @allocator.read_pointer, keys, num_keys)
|
125
|
+
num_keys.read(:int64_t).times do |i|
|
126
|
+
key = keys.read_pointer[i * ::FFI::Pointer.size].read_pointer.read_string
|
127
|
+
value = ::FFI::MemoryPointer.new(:string)
|
128
|
+
check_status api[:ModelMetadataLookupCustomMetadataMap].call(metadata.read_pointer, @allocator.read_pointer, key, value)
|
129
|
+
custom_metadata_map[key] = value.read_pointer.read_string
|
130
|
+
end
|
131
|
+
|
120
132
|
check_status api[:ModelMetadataGetDescription].call(metadata.read_pointer, @allocator.read_pointer, description)
|
121
133
|
check_status api[:ModelMetadataGetDomain].call(metadata.read_pointer, @allocator.read_pointer, domain)
|
122
134
|
check_status api[:ModelMetadataGetGraphName].call(metadata.read_pointer, @allocator.read_pointer, graph_name)
|
@@ -124,10 +136,8 @@ module OnnxRuntime
|
|
124
136
|
check_status api[:ModelMetadataGetVersion].call(metadata.read_pointer, version)
|
125
137
|
api[:ReleaseModelMetadata].call(metadata.read_pointer)
|
126
138
|
|
127
|
-
# TODO add custom_metadata_map
|
128
|
-
# need a way to get keys
|
129
|
-
|
130
139
|
{
|
140
|
+
custom_metadata_map: custom_metadata_map,
|
131
141
|
description: description.read_pointer.read_string,
|
132
142
|
domain: domain.read_pointer.read_string,
|
133
143
|
graph_name: graph_name.read_pointer.read_string,
|
data/lib/onnxruntime/version.rb
CHANGED
data/vendor/libonnxruntime.dylib
CHANGED
Binary file
|
data/vendor/libonnxruntime.so
CHANGED
Binary file
|
data/vendor/onnxruntime.dll
CHANGED
Binary file
|
metadata
CHANGED
@@ -1,14 +1,14 @@
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
2
2
|
name: onnxruntime
|
3
3
|
version: !ruby/object:Gem::Version
|
4
|
-
version: 0.3.
|
4
|
+
version: 0.3.1
|
5
5
|
platform: ruby
|
6
6
|
authors:
|
7
7
|
- Andrew Kane
|
8
8
|
autorequire:
|
9
9
|
bindir: bin
|
10
10
|
cert_chain: []
|
11
|
-
date: 2020-
|
11
|
+
date: 2020-05-19 00:00:00.000000000 Z
|
12
12
|
dependencies:
|
13
13
|
- !ruby/object:Gem::Dependency
|
14
14
|
name: ffi
|