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