:github_url: https://github.com/pytorch/pytorch .. _exhale_function_namespaceat_1a99894de13fe7feccc3bcf084ee870f02: Function at::_scaled_dot_product_flash_attention ================================================ - Defined in :ref:`file_build_aten_src_ATen_Functions.h` Function Documentation ---------------------- .. doxygenfunction:: at::_scaled_dot_product_flash_attention(const at::Tensor&, const at::Tensor&, const at::Tensor&, double, bool, bool, ::std::optional) :project: PyTorch