Searched defs:proj_bias (Results 1 – 4 of 4) sorted by relevance
82 const Tensor& proj_bias, in transformer_encoder_layer_forward()
270 const Tensor& proj_bias, in native_multi_head_attention_cpu()861 const Tensor& proj_bias, in triton_multi_head_attention()
488 const Tensor& proj_bias, in native_multi_head_attention_cuda()
300 proj_bias, argument