Skip to content

Commit

Permalink
New types added
Browse files Browse the repository at this point in the history
  • Loading branch information
ayush1999 committed Feb 13, 2018
1 parent 2e507c6 commit a235bba
Showing 1 changed file with 40 additions and 0 deletions.
40 changes: 40 additions & 0 deletions src/new_types.jl
@@ -0,0 +1,40 @@
#=
Now, we define the new data types.
Model => ModelProto
Graph => GraphProto
Node => NodeProto
Attribute => AttributeProto
The new types will consist of Julian attributes.
=#

mutable struct Node
input::Vector{AbstractString}
output::Vector{AbstractString}
name::AbstractString
op_type::AbstractString
domain::AbstractString
attribute::Dict{Any, Array{Any, 1}} #AttributeProto to Dict
doc_string::AbstractString
end

mutable struct Graph
node::Vector{Node}
name::AbstractString
initializer::Dict{Any, Array{Any, 1}} #Storing the array data instead of the tensorproto vector.
doc_string::AbstractString #in Dict format.
input::Vector{Dict} #
output::Vector{Dict} # ValueInfoProto to Dict type.
value_info::Vector{Dict} #
end

mutable struct Model
ir_version::Int64
opset_import::Vector{Dict} #OperatorSetIdProto to Dict
producer_name::AbstractString
producer_version::AbstractString
domain::AbstractString
model_version::Int64
doc_string::AbstractString
graph::Graph
metadata_props::Vector{Dict} #StringStringEntryProto to Dict
end

0 comments on commit a235bba

Please sign in to comment.