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