include/ck_tile/ops/softmax.hpp Source File

• include
• ck_tile
• ops

softmax.hpp

Go to the documentation of this file.

1 // SPDX-License-Identifier: MIT

2 // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.

3 

4 #pragma once

5 

6 #include "ck_tile/ops/softmax/block/block_softmax_2d.hpp"

7 #include "ck_tile/ops/softmax/block/block_softmax_2d_problem.hpp"

8 #include "ck_tile/ops/common/generic_2d_block_shape.hpp"

9 #include "ck_tile/ops/common/tensor_layout.hpp"

block_softmax_2d.hpp

block_softmax_2d_problem.hpp

generic_2d_block_shape.hpp

tensor_layout.hpp