Searched defs:decoder_input (Results 1 – 1 of 1) sorted by relevance
258 torch::Tensor decoder_input = in transformer_decoder_layer_test_helper() local470 torch::Tensor decoder_input = in transformer_decoder_layer_test_helper_gelu() local856 torch::Tensor decoder_input = in transformer_decoder_test_helper() local