Sha256: 45315249298b443077dad336565cbac76c4a95a1c99ecc56a3d75414699d6d54

Contents?: true

Size: 395 Bytes

Versions: 1

Compression:

Stored size: 395 Bytes

Contents

require 'chainer'
require "onnx-chainer/version"
require "onnx-chainer/graph"
require "onnx-chainer/proto/onnx_pb"

module OnnxChainer
  class Error < StandardError; end

  def self.parse_file(onnx_path)
    raise "File not found. #{onnx_path}" if onnx_path.nil? || !File.exists?(onnx_path)

    m = Onnx::ModelProto.decode(File.read(onnx_path))

    OnnxChainer::Graph.parse(m.graph)
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version Path
onnx-red-chainer-0.1.0 lib/onnx-chainer.rb