Searched defs:q_t (Results 1 – 5 of 5) sorted by relevance
| /external/clang/test/PCH/ |
| D | ocl_types.h | 33 typedef queue_t q_t; typedef
|
| /external/pytorch/aten/src/ATen/native/nested/cuda/ |
| D | NestedTensorTransformerUtils.cpp | 232 Tensor q_t = query.transpose(1, 2); in sdpa_nested_preprocessing_with_broadcast() local
409 Tensor q_t = query.transpose(1, 2); in sdpa_nested_preprocessing() local
486 Tensor q_t = query.transpose(1, 2); in sdpa_nested_preprocessing_backward() local
|
| /external/pytorch/aten/src/ATen/native/transformers/cuda/ |
| D | attention_backward.cu | 431 at::Tensor q_t = query.permute({0,2,1,3}); in _efficient_attention_backward() local
752 Tensor q_t = query.transpose(1, 2); in _scaled_dot_product_flash_attention_backward_cuda() local
804 auto q_t = query.transpose(1, 2); in _scaled_dot_product_efficient_attention_backward_cuda() local
|
| D | attention.cu | 708 Tensor q_t = query.transpose(1, 2); in _scaled_dot_product_flash_attention_cuda() local
803 Tensor q_t = query.transpose(1, 2); in _scaled_dot_product_efficient_attention_cuda() local
1090 at::Tensor q_t = query.transpose(1, 2); in _efficient_attention_forward() local
|
| /external/pytorch/aten/src/ATen/native/transformers/ |
| D | attention.cpp | 831 auto q_t = query.transpose(1, 2); in _scaled_dot_product_flash_attention_cpu_backward() local
|