Searched defs:query_bdim (Results 1 – 1 of 1) sorted by relevance
491 const Tensor& query, std::optional<int64_t> query_bdim, in _scaled_dot_product_flash_attention_batch_rule()539 const Tensor& query, optional<int64_t> query_bdim, in _scaled_dot_product_efficient_attention_batch_rule()580 const Tensor& query, std::optional<int64_t> query_bdim, in _scaled_dot_product_cudnn_attention_batch_rule()