Searched defs:decoder_input (Results 1 – 2 of 2) 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
69 int decoder_input = 0; in main() local