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 CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA256:
3
- metadata.gz: 1a31036492f6ad4f0ab17a2534a69f17ca05058267447089868f2ac5bf2c5e79
4
- data.tar.gz: 6cd8be229992a0ea4c9fdb8529a229e31a777321e0971f0123bc2680ca828be3
3
+ metadata.gz: 2d4b9ddc710644eab07227fe005c92ad039643e989236e5f112a83df2266fc55
4
+ data.tar.gz: d15c9ac4ae98130499d1411d3be2e7b822df8248550c0ad44c2e7a02d88f95d8
5
5
  SHA512:
6
- metadata.gz: 9bb61d56c9c4ddb17e0e085381226ab0c91e72de592deb6519bf6840ffb1a00eaa0fb9382eb0cba1d563165fb9d0528034da1f2ea9b6b3cb76b780813750ad21
7
- data.tar.gz: b94b0b2772728ead241a0d16c3ca2db099cbf7ab6f3994384e3107f69d9aa2cdaa31ed4ba08bd4d3d9fae2eef0b13ea84aa70823be0ff2d24b921e8fb390246b
6
+ metadata.gz: 83a2e3699506460b8b280376ec8935a680e7e2cffe6bf7dd6766a85eea84bf06273e13889b6a2961c9df3c277be6ea7d8afbb67a4d4be62d50b266d7ccd8dc55
7
+ data.tar.gz: 5086c2c04f76ec67224e7dbc9ccc36366b394840296536232b29d92009c7a4ef637fb4f39daec8abb7a5e3fdcca591675cfaf3e4a8e31ec939f9aa1e2ae6f112
data/CHANGELOG.md CHANGED
@@ -1,3 +1,8 @@
1
+ ## 0.3.1 (2020-05-18)
2
+
3
+ - Updated ONNX Runtime to 1.3.0
4
+ - Added `custom_metadata_map` to model metadata
5
+
1
6
  ## 0.3.0 (2020-03-11)
2
7
 
3
8
  - Updated ONNX Runtime to 1.2.0
@@ -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,
@@ -1,3 +1,3 @@
1
1
  module OnnxRuntime
2
- VERSION = "0.3.0"
2
+ VERSION = "0.3.1"
3
3
  end
Binary file
Binary file
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.0
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-03-11 00:00:00.000000000 Z
11
+ date: 2020-05-19 00:00:00.000000000 Z
12
12
  dependencies:
13
13
  - !ruby/object:Gem::Dependency
14
14
  name: ffi