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)
263private protected override Delegate CreateGetter(DataViewRow input, int iinfo, TensorCacher outputCacher) 269private Delegate MakeScoreGetter(DataViewRow input, IChannel ch, TensorCacher outputCacher)
NasBert\TextClassificationTrainer.cs (3)
306private protected override Delegate CreateGetter(DataViewRow input, int iinfo, TensorCacher outputCacher) 315private Delegate MakeScoreGetter(DataViewRow input, IChannel ch, TensorCacher outputCacher) 346private 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();