1 type derived from TensorCacher
Microsoft.ML.TorchSharp (1)
TorchSharpBaseTrainer.cs (1)
530private protected abstract class TensorCacher<TOut> : TensorCacher
13 references to TensorCacher
Microsoft.ML.TorchSharp (13)
NasBert\NasBertTrainer.cs (2)
577private protected override TensorCacher GetTensorCacher() 612private 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)
370private protected override Delegate CreateGetter(DataViewRow input, int iinfo, TensorCacher outputCacher) 377private void CondenseOutput(ref VBuffer<UInt32> dst, string sentence, Tokenizer tokenizer, TensorCacher outputCacher) 412private Delegate MakePredictedLabelGetter(DataViewRow input, IChannel ch, TensorCacher outputCacher)
NasBert\SentenceSimilarityTrainer.cs (2)
264private protected override Delegate CreateGetter(DataViewRow input, int iinfo, TensorCacher outputCacher) 270private Delegate MakeScoreGetter(DataViewRow input, IChannel ch, TensorCacher outputCacher)
NasBert\TextClassificationTrainer.cs (3)
307private protected override Delegate CreateGetter(DataViewRow input, int iinfo, TensorCacher outputCacher) 316private Delegate MakeScoreGetter(DataViewRow input, IChannel ch, TensorCacher outputCacher) 347private Delegate MakePredictedLabelGetter(DataViewRow input, IChannel ch, TensorCacher outputCacher)
TorchSharpBaseTrainer.cs (3)
488private protected abstract Delegate CreateGetter(DataViewRow input, int iinfo, TensorCacher outputCacher); 495TensorCacher outputCacher = GetTensorCacher(); 514private protected abstract TensorCacher GetTensorCacher();