1 instantiation of BertTensorCacher
Microsoft.ML.TorchSharp (1)
NasBert\NasBertTrainer.cs (1)
579
return new
BertTensorCacher
();
8 references to BertTensorCacher
Microsoft.ML.TorchSharp (8)
NasBert\NasBertTrainer.cs (2)
614
var
cache = outputCache as
BertTensorCacher
;
NasBert\NerTrainer.cs (2)
382
var argmax = (outputCacher as
BertTensorCacher
).Result.argmax(-1);
428
var argmax = (outputCacher as
BertTensorCacher
).Result.argmax(-1);
NasBert\SentenceSimilarityTrainer.cs (2)
281
var
cacher = outputCacher as
BertTensorCacher
;
NasBert\TextClassificationTrainer.cs (2)
334
var values = (outputCacher as
BertTensorCacher
).Result.cpu().ToArray<float>();
364
dst = (UInt32)(outputCacher as
BertTensorCacher
).Result.argmax(-1).cpu().item<long>() + 1;