1 instantiation of TensorCacher
Microsoft.ML.TorchSharp (1)
AutoFormerV2\ObjectDetectionTrainer.cs (1)
881TensorCacher outputCacher = new TensorCacher(_parent.Options.NumberOfClasses);
6 references to TensorCacher
Microsoft.ML.TorchSharp (6)
AutoFormerV2\ObjectDetectionTrainer.cs (6)
793private Delegate CreateGetter(DataViewRow input, int iinfo, TensorCacher outputCacher) 804private Delegate MakeScoreGetter(DataViewRow input, IChannel ch, TensorCacher outputCacher) 828private Delegate MakePredictedLabelGetter(DataViewRow input, IChannel ch, TensorCacher outputCacher) 852private Delegate MakeBoundingBoxGetter(DataViewRow input, IChannel ch, TensorCacher outputCacher) 881TensorCacher outputCacher = new TensorCacher(_parent.Options.NumberOfClasses); 963private protected void UpdateCacheIfNeeded(long position, TensorCacher outputCache, ref MLImage image, ref ValueGetter<MLImage> getImage)