Searched defs:s_q (Results 1 – 2 of 2) sorted by relevance
14 int64_t s_q, in run_cudnn_SDP_fprop()36 int64_t s_q, in run_cudnn_SDP_bprop()131 int64_t s_q; member144 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()
314 const auto s_q = params.query.sym_size(2); in check_cudnn_tensor_shapes() local371 const int64_t s_q = params.query.size(2); in check_cudnn_layout() local