Searched defs:use_callable_activation (Results 1 – 1 of 1) sorted by relevance
29 bool use_callable_activation) { in get_a_test_layer()56 bool use_callable_activation) { in transformer_encoder_layer_test_helper()248 bool use_callable_activation) { in transformer_decoder_layer_test_helper()454 bool use_callable_activation) { in transformer_decoder_layer_test_helper_gelu()567 bool use_callable_activation) { in transformer_encoder_test_helper()841 bool use_callable_activation) { in transformer_decoder_test_helper()1366 void transformer_test_helper(bool is_cuda, bool use_callable_activation) { in transformer_test_helper()