Home
last modified time | relevance | path

Searched defs:s_q (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cudnn/
DMHA.cpp14 int64_t s_q, in run_cudnn_SDP_fprop()
36 int64_t s_q, in run_cudnn_SDP_bprop()
131 int64_t s_q; member
144 int64_t s_q, in setMHAParams()
257 int64_t s_q, in build_graph_and_tensors()
392 int64_t s_q, in build_graph_and_tensors_backward()
524 int64_t s_q, in run_cudnn_SDP_fprop()
612 int64_t s_q, in run_cudnn_SDP_bprop()
/external/pytorch/aten/src/ATen/native/transformers/cuda/
Dsdp_utils.cpp314 const auto s_q = params.query.sym_size(2); in check_cudnn_tensor_shapes() local
371 const int64_t s_q = params.query.size(2); in check_cudnn_layout() local