1 instantiation of TensorCacher
Microsoft.ML.TorchSharp (1)
Roberta\QATrainer.cs (1)
832TensorCacher outputCacher = new TensorCacher(_parent.Options.TopKAnswers);
5 references to TensorCacher
Microsoft.ML.TorchSharp (5)
Roberta\QATrainer.cs (5)
764private Delegate CreateGetter(DataViewRow input, int iinfo, TensorCacher outputCacher) 773private Delegate MakeScoreGetter(DataViewRow input, IChannel ch, TensorCacher outputCacher) 800private Delegate MakePredictedAnswerGetter(DataViewRow input, IChannel ch, TensorCacher outputCacher) 832TensorCacher outputCacher = new TensorCacher(_parent.Options.TopKAnswers); 909private protected void UpdateCacheIfNeeded(long position, TensorCacher outputCache, ref ReadOnlyMemory<char> context, ref ReadOnlyMemory<char> question, ref ValueGetter<ReadOnlyMemory<char>> getContext, ref ValueGetter<ReadOnlyMemory<char>> getQuestion)