Home
last modified time | relevance | path

Searched defs:proj_weight (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/
H A Dattention.cpp269 const Tensor& proj_weight, in native_multi_head_attention_cpu()
860 const Tensor& proj_weight, in triton_multi_head_attention()
H A Dtransformer.cpp81 const Tensor& proj_weight, in transformer_encoder_layer_forward()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/
H A Dattention.cu487 const Tensor& proj_weight, in native_multi_head_attention_cuda()
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_cpu_repro.py299 proj_weight, argument