/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/grouped_convolution.hpp Source File

• develop
• include
• ck_tile
• ops

grouped_convolution.hpp

Go to the documentation of this file.

1 // Copyright (c) Advanced Micro Devices, Inc., or its affiliates.

2 // SPDX-License-Identifier: MIT

3 #pragma once

4 

5 #include "ck_tile/ops/grouped_convolution/kernel/grouped_convolution_backward_data_kernel.hpp"

6 #include "ck_tile/ops/grouped_convolution/kernel/grouped_convolution_backward_weight_kernel.hpp"

7 #include "ck_tile/ops/grouped_convolution/kernel/grouped_convolution_forward_kernel.hpp"

8 #include "ck_tile/ops/grouped_convolution/utils/convolution_specialization.hpp"

9 #include "ck_tile/ops/grouped_convolution/utils/grouped_convolution_utils.hpp"

10 #include "ck_tile/ops/grouped_convolution/utils/split_k_utils.hpp"

11 #include "ck_tile/ops/grouped_convolution/utils/transform_conv_bwd_data_to_gemm.hpp"

12 #include "ck_tile/ops/grouped_convolution/utils/transform_conv_bwd_weight_to_gemm.hpp"

13 #include "ck_tile/ops/grouped_convolution/utils/transform_conv_fwd_to_gemm.hpp"

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

15 #include "ck_tile/ops/common/load_interleaved_pk_type.hpp"

16 #include "ck_tile/ops/common/streamk_common.hpp"

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

18 #include "ck_tile/ops/common/utils.hpp"

convolution_specialization.hpp

generic_2d_block_shape.hpp

grouped_convolution_backward_data_kernel.hpp

grouped_convolution_backward_weight_kernel.hpp

grouped_convolution_forward_kernel.hpp

grouped_convolution_utils.hpp

load_interleaved_pk_type.hpp

streamk_common.hpp

tensor_layout.hpp

split_k_utils.hpp

transform_conv_bwd_weight_to_gemm.hpp

transform_conv_fwd_to_gemm.hpp

transform_conv_bwd_data_to_gemm.hpp

utils.hpp