/// <summary> /// Makes the ONNX model based on the context. /// </summary> public OnnxCSharpToProtoWrapper.ModelProto MakeModel() => OnnxUtils.MakeModel(_nodes, _producerName, _name, _domain, _producerVersion, _modelVersion, _inputs, _outputs, _intermediateValues, _initializers);
/// Adds constant tensor into the graph. public override string AddInitializer(float value, string name = null, bool makeUniqueName = true) { name = AddVariable(name ?? "float", makeUniqueName); _initializers.Add(OnnxUtils.MakeFloat(name, value)); return(name); }
public override string AddInitializer(long value, string name = null, bool makeUniqueName = true) { name = AddVariable(name ?? "int64", makeUniqueName); _initializers.Add(OnnxUtils.MakeInt64(name, value)); return(name); }
/// <summary> /// Adds an output variable to the list. /// </summary> public void AddOutputVariable(DataViewType type, string variableName, List <long> dim = null) { _host.CheckValue(type, nameof(type)); _host.CheckParam(IsVariableDefined(variableName), nameof(variableName)); _outputs.Add(OnnxUtils.GetModelArgs(type, variableName, dim)); }
public override string AddInitializer(string value, string name = null) { name = AddVariable(name ?? "string"); _initializers.Add(OnnxUtils.MakeString(name, value)); return(name); }
public override void AddAttribute(string argName, bool value) => OnnxUtils.NodeAddAttributes(_node, argName, value);
public override void AddAttribute(string argName, IEnumerable <string> value) => OnnxUtils.NodeAddAttributes(_node, argName, value);
public override void AddAttribute(string argName, IEnumerable <ReadOnlyMemory <char> > value) => OnnxUtils.NodeAddAttributes(_node, argName, value);