Searched defs:proj_weight (Results 1 – 4 of 4) sorted by relevance
81 const Tensor& proj_weight, in transformer_encoder_layer_forward()
269 const Tensor& proj_weight, in native_multi_head_attention_cpu()860 const Tensor& proj_weight, in triton_multi_head_attention()
487 const Tensor& proj_weight, in native_multi_head_attention_cuda()
299 proj_weight, argument