/// <summary> /// Adds an input variable to the list. /// </summary> public void AddInputVariable(DataViewType type, string colName) { _host.CheckValue(type, nameof(type)); _host.CheckValue(colName, nameof(colName)); colName = AddVariable(colName); _inputs.Add(OnnxUtils.GetModelArgs(type, colName)); }
/// <summary> /// Adds an intermediate column to the list. /// </summary> public override string AddIntermediateVariable(DataViewType type, string colName, bool skip = false) { colName = AddVariable(colName); // Let the runtime figure the shape. if (!skip) { _host.CheckValue(type, nameof(type)); _intermediateValues.Add(OnnxUtils.GetModelArgs(type, colName)); } return(colName); }
/// <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)); }