↓ 4 callersMethod_beam_search_generate(
self,
model,
input_ids,
attention_mask,
max_length,
beam_sco
src/transformers/tests/generation/test_utils.py:384
↓ 4 callersMethod_expand_inputs_for_generation Expands tensors from [batch_size, ...] to [batch_size * expand_size, ...] or [batch_size, expand_size, ...], depending on `expand_in_
src/transformers/src/transformers/generation/tf_utils.py:1135
↓ 4 callersMethodcheck_iterable_dataset_shard(self, dataset, batch_size, drop_last, num_processes=2, epoch=0)
src/transformers/tests/trainer/test_trainer_utils.py:303
↓ 4 callersFunctionevaluate(args, model, tokenizer, prefix="", output_layer=-1, eval_highway=False)
src/transformers/examples/research_projects/deebert/run_glue_deebert.py:240