1 type derived from TensorCacher
Microsoft.ML.TorchSharp (1)
TorchSharpBaseTrainer.cs (1)
530
private protected abstract class TensorCacher<TOut> :
TensorCacher
13 references to TensorCacher
Microsoft.ML.TorchSharp (13)
NasBert\NasBertTrainer.cs (2)
577
private protected override
TensorCacher
GetTensorCacher()
612
private protected void UpdateCacheIfNeeded(long position,
TensorCacher
outputCache, ref ReadOnlyMemory<char> sentence1, ref ReadOnlyMemory<char> sentence2, ref ValueGetter<ReadOnlyMemory<char>> getSentence1, ref ValueGetter<ReadOnlyMemory<char>> getSentence2, Tokenizer tokenizer)
NasBert\NerTrainer.cs (3)
370
private protected override Delegate CreateGetter(DataViewRow input, int iinfo,
TensorCacher
outputCacher)
377
private void CondenseOutput(ref VBuffer<UInt32> dst, string sentence, Tokenizer tokenizer,
TensorCacher
outputCacher)
412
private Delegate MakePredictedLabelGetter(DataViewRow input, IChannel ch,
TensorCacher
outputCacher)
NasBert\SentenceSimilarityTrainer.cs (2)
263
private protected override Delegate CreateGetter(DataViewRow input, int iinfo,
TensorCacher
outputCacher)
269
private Delegate MakeScoreGetter(DataViewRow input, IChannel ch,
TensorCacher
outputCacher)
NasBert\TextClassificationTrainer.cs (3)
306
private protected override Delegate CreateGetter(DataViewRow input, int iinfo,
TensorCacher
outputCacher)
315
private Delegate MakeScoreGetter(DataViewRow input, IChannel ch,
TensorCacher
outputCacher)
346
private Delegate MakePredictedLabelGetter(DataViewRow input, IChannel ch,
TensorCacher
outputCacher)
TorchSharpBaseTrainer.cs (3)
488
private protected abstract Delegate CreateGetter(DataViewRow input, int iinfo,
TensorCacher
outputCacher);
495
TensorCacher
outputCacher = GetTensorCacher();
514
private protected abstract
TensorCacher
GetTensorCacher();