Module: OnnxRuby::Model::ClassMethods

Defined in:
lib/onnx_ruby/model.rb

Instance Method Summary collapse

Instance Method Details

#onnx_input(callable = nil, &block) ⇒ Object 



31
32
33
34
 
# File 'lib/onnx_ruby/model.rb', line 31

def onnx_input(callable = nil, &block)
  @onnx_input_fn = callable || block if callable || block
  @onnx_input_fn
end

#onnx_model(path = nil, **opts) ⇒ Object 



23
24
25
26
27
28
29
 
# File 'lib/onnx_ruby/model.rb', line 23

def onnx_model(path = nil, **opts)
  if path
    @onnx_model_path = path
    @onnx_session_opts = opts
  end
  @onnx_model_path
end

#onnx_output(name = nil) ⇒ Object 



36
37
38
39
 
# File 'lib/onnx_ruby/model.rb', line 36

def onnx_output(name = nil)
  @onnx_output_name = name if name
  @onnx_output_name
end

#onnx_session ⇒ Object 



41
42
43
44
45
46
 
# File 'lib/onnx_ruby/model.rb', line 41

def onnx_session
  @onnx_session ||= begin
    path = resolve_model_path(@onnx_model_path)
    LazySession.new(path, **(@onnx_session_opts || {}))
  end
end