87 references to OnnxContext
Microsoft.ML.Data (83)
DataView\RowToRowMapperTransform.cs (2)
91bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => _mapper is ICanSaveOnnx onnxMapper ? onnxMapper.CanSaveOnnx(ctx) : false; 217void ISaveAsOnnx.SaveAsOnnx(OnnxContext ctx)
Model\Onnx\ICanSaveOnnx.cs (5)
20bool CanSaveOnnx(OnnxContext ctx); 33void SaveAsOnnx(OnnxContext ctx); 67bool SaveAsOnnx(OnnxContext ctx, RoleMappedSchema schema, string[] outputNames); 77bool SaveAsOnnx(OnnxContext ctx, string[] outputNames, string featureColumn); 87new bool SaveAsOnnx(OnnxContext ctx, string[] outputNames, string featureColumn);
Model\Onnx\OnnxContext.cs (2)
119/// which ought to have been something returned from <see cref="OnnxContext.AddIntermediateVariable(DataViewType, string, bool)"/></param> 120/// <param name="name">The name of the operator, which ought to be something returned from <see cref="OnnxContext.GetNodeName(string)"/></param>
Model\Onnx\OnnxNode.cs (1)
12/// <see cref="OnnxContext.CreateNode(string, IEnumerable{string}, IEnumerable{string}, string, string)"/>.
Prediction\Calibrator.cs (9)
305bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => (_mapper as ICanSaveOnnx)?.CanSaveOnnx(ctx) == true; 376bool IDistCanSaveOnnx.SaveAsOnnx(OnnxContext ctx, string[] outputNames, string featureColumnName) 379bool ISingleCanSaveOnnx.SaveAsOnnx(OnnxContext ctx, string[] outputNames, string featureColumnName) 750bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => (_bindable as ICanSaveOnnx)?.CanSaveOnnx(ctx) == true; 791bool IBindableCanSaveOnnx.SaveAsOnnx(OnnxContext ctx, RoleMappedSchema schema, string[] outputs) 1186bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => true; 1294bool ISingleCanSaveOnnx.SaveAsOnnx(OnnxContext ctx, string[] outputNames, string featureColumn) 1702bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => true; 1792bool ISingleCanSaveOnnx.SaveAsOnnx(OnnxContext ctx, string[] scoreProbablityColumnNames, string featureColumnName)
Prediction\CalibratorCatalog.cs (2)
247bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => _calibrator is ICanSaveOnnx onnxMapper ? onnxMapper.CanSaveOnnx(ctx) : false; 266void ISaveAsOnnx.SaveAsOnnx(OnnxContext ctx)
Scorers\BinaryClassifierScorer.cs (1)
184private protected override void SaveAsOnnxCore(OnnxContext ctx)
Scorers\GenericScorer.cs (2)
147bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => (Bindable as ICanSaveOnnx)?.CanSaveOnnx(ctx) == true; 225void ISaveAsOnnx.SaveAsOnnx(OnnxContext ctx)
Scorers\MulticlassClassificationScorer.cs (2)
91bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => (_bindable as ICanSaveOnnx)?.CanSaveOnnx(ctx) == true; 220bool IBindableCanSaveOnnx.SaveAsOnnx(OnnxContext ctx, RoleMappedSchema schema, string[] outputNames)
Scorers\PredictedLabelScorerBase.cs (3)
273bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => (Bindable as ICanSaveOnnx)?.CanSaveOnnx(ctx) == true; 360void ISaveAsOnnx.SaveAsOnnx(OnnxContext ctx) => SaveAsOnnxCore(ctx); 363private protected virtual void SaveAsOnnxCore(OnnxContext ctx)
Scorers\SchemaBindablePredictorWrapper.cs (5)
49bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => (ValueMapper as ICanSaveOnnx)?.CanSaveOnnx(ctx) == true; 108bool IBindableCanSaveOnnx.SaveAsOnnx(OnnxContext ctx, RoleMappedSchema schema, string[] outputNames) 118private protected virtual bool SaveAsOnnxCore(OnnxContext ctx, RoleMappedSchema schema, string[] outputNames) => false; 340private protected override bool SaveAsOnnxCore(OnnxContext ctx, RoleMappedSchema schema, string[] outputNames) 451private protected override bool SaveAsOnnxCore(OnnxContext ctx, RoleMappedSchema schema, string[] outputNames)
Transforms\ColumnConcatenatingTransformer.cs (2)
425public bool CanSaveOnnx(OnnxContext ctx) => true; 900public void SaveAsOnnx(OnnxContext ctx)
Transforms\ColumnCopying.cs (2)
204public bool CanSaveOnnx(OnnxContext ctx) => true; 238public void SaveAsOnnx(OnnxContext ctx)
Transforms\ColumnSelecting.cs (2)
731public bool CanSaveOnnx(OnnxContext ctx) => true; 733public void SaveAsOnnx(OnnxContext ctx)
Transforms\Hashing.cs (3)
1356private bool SaveAsOnnxCore(OnnxContext ctx, int iinfo, string srcVariable, string dstVariable) 1456void ISaveAsOnnx.SaveAsOnnx(OnnxContext ctx) 1476bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => true;
Transforms\KeyToValue.cs (4)
303public abstract bool SaveOnnx(OnnxContext ctx, string srcVariableName, string dstVariableName); 501public override bool SaveOnnx(OnnxContext ctx, string srcVariableName, string dstVariableName) 578public bool CanSaveOnnx(OnnxContext ctx) => true; 580public void SaveAsOnnx(OnnxContext ctx)
Transforms\KeyToVector.cs (3)
610public bool CanSaveOnnx(OnnxContext ctx) => true; 614public void SaveAsOnnx(OnnxContext ctx) 704private void SaveAsOnnxCore(OnnxContext ctx, int iinfo, ColInfo info, string srcVariableName, string dstVariableName)
Transforms\NormalizeColumn.cs (6)
421public bool CanSaveOnnx(OnnxContext ctx) => true; 422public abstract bool OnnxInfo(OnnxContext ctx, OnnxNode nodeProtoWrapper, int featureCount); 540public bool CanSaveOnnx(OnnxContext ctx) => false; 542public bool OnnxInfo(OnnxContext ctx, OnnxNode nodeProtoWrapper, int featureCount) 671public bool CanSaveOnnx(OnnxContext ctx) => false; 673public bool OnnxInfo(OnnxContext ctx, OnnxNode nodeProtoWrapper, int featureCount)
Transforms\NormalizeColumnDbl.cs (2)
700public override bool OnnxInfo(OnnxContext ctx, OnnxNode nodeProtoWrapper, int featureCount) 772public override bool OnnxInfo(OnnxContext ctx, OnnxNode node, int featureCount)
Transforms\NormalizeColumnSng.cs (2)
858public override bool OnnxInfo(OnnxContext ctx, OnnxNode node, int featureCount) 929public override bool OnnxInfo(OnnxContext ctx, OnnxNode node, int featureCount)
Transforms\Normalizer.cs (3)
724public bool CanSaveOnnx(OnnxContext ctx) => true; 762public void SaveAsOnnx(OnnxContext ctx) 823private bool SaveAsOnnxCore(OnnxContext ctx, int iinfo, ColumnOptions info, string srcVariableName, string dstVariableName)
Transforms\NormalizeUtils.cs (2)
66bool CanSaveOnnx(OnnxContext ctx); 68bool OnnxInfo(OnnxContext ctx, OnnxNode nodeProtoWrapper, int featureCount);
Transforms\SlotsDroppingTransformer.cs (3)
877public bool CanSaveOnnx(OnnxContext ctx) => true; 879public void SaveAsOnnx(OnnxContext ctx) 896public bool SaveAsOnnxCore(OnnxContext ctx, int iinfo, string srcVariableName, string dstVariableName)
Transforms\TransformBase.cs (3)
493bool ICanSaveOnnx.CanSaveOnnx(OnnxContext ctx) => CanSaveOnnxCore; 603void ISaveAsOnnx.SaveAsOnnx(OnnxContext ctx) 650private protected virtual bool SaveAsOnnxCore(OnnxContext ctx, int iinfo, ColInfo info, string srcVariableName,
Transforms\TypeConverting.cs (3)
390public bool CanSaveOnnx(OnnxContext ctx) => true; 477public void SaveAsOnnx(OnnxContext ctx) 498private bool SaveAsOnnxCore(OnnxContext ctx, int iinfo, string srcVariableName, string dstVariableName)
Transforms\ValueMapping.cs (4)
1034public bool CanSaveOnnx(OnnxContext ctx) => true; 1057public void SaveAsOnnx(OnnxContext ctx) 1085private void CastInputTo<T>(OnnxContext ctx, out OnnxNode node, string srcVariableName, string opType, string labelEncoderOutput, PrimitiveDataViewType itemType) 1101private bool SaveAsOnnxCore(OnnxContext ctx, string srcVariableName, string dstVariableName)
Transforms\ValueToKeyMappingTransformer.cs (5)
712public bool CanSaveOnnx(OnnxContext ctx) => true; 787private void CastInputToString<T>(OnnxContext ctx, out OnnxNode node, out long[] termIds, string srcVariableName, int iinfo, 800private void CastInputToFloat<T>(OnnxContext ctx, out OnnxNode node, out long[] termIds, string srcVariableName, int iinfo, 813private bool SaveAsOnnxCore(OnnxContext ctx, int iinfo, ColInfo info, string srcVariableName, string dstVariableName) 902public void SaveAsOnnx(OnnxContext ctx)
Microsoft.ML.StandardTrainers (4)
Standard\LogisticRegression\MulticlassLogisticRegression.cs (4)
974/// Actual implementation of <see cref="ISingleCanSaveOnnx.SaveAsOnnx(OnnxContext, string[], string)"/> should happen in derived classes. 975/// It's ok to make <see cref="SaveAsOnnxCore(OnnxContext, string[], string)"/> a <see langword="private protected"/> method in the future 1013/// a softmax function. This function is used only in <see cref="SaveAsOnnxCore(OnnxContext, string[], string)"/>. 1018/// Redirect <see cref="ISingleCanSaveOnnx.SaveAsOnnx(OnnxContext, string[], string)"/> call to the right function.