2 references to CollateTokens
Microsoft.ML.TorchSharp (2)
NasBert\NasBertTrainer.cs (1)
240
return DataUtils.
CollateTokens
(inputTensors, Tokenizer.RobertaModel().PadIndex, device: Device);
Roberta\QATrainer.cs (1)
386
return DataUtils.
CollateTokens
(inputTensors, padIndex, device: Device);