Class: OnnxRuby::LazySession

Inherits:
Object
  • Object
show all
Defined in:
lib/onnx_ruby/lazy_session.rb

Instance Method Summary collapse

Constructor Details

#initialize(model_path, **opts) ⇒ LazySession

Returns a new instance of LazySession.



5
6
7
8
9
10
 
# File 'lib/onnx_ruby/lazy_session.rb', line 5

def initialize(model_path, **opts)
  @model_path = model_path
  @opts = opts
  @session = nil
  @mutex = Mutex.new
end

Instance Method Details

#inputs ⇒ Object 



12
13
14
 
# File 'lib/onnx_ruby/lazy_session.rb', line 12

def inputs
  load_session.inputs
end

#loaded? ⇒ Boolean

Returns:

  • (Boolean) 


24
25
26
 
# File 'lib/onnx_ruby/lazy_session.rb', line 24

def loaded?
  !@session.nil?
end

#outputs ⇒ Object 



16
17
18
 
# File 'lib/onnx_ruby/lazy_session.rb', line 16

def outputs
  load_session.outputs
end

#run(inputs, **kwargs) ⇒ Object 



20
21
22
 
# File 'lib/onnx_ruby/lazy_session.rb', line 20

def run(inputs, **kwargs)
  load_session.run(inputs, **kwargs)
end