/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-7.0.0/include/ck_tile/ops/grouped_convolution/kernel/grouped_convolution_forward_kernel.hpp File Reference#
grouped_convolution_forward_kernel.hpp File Reference
#include <iostream>#include <string>#include "ck_tile/core.hpp"#include "ck_tile/ops/common.hpp"#include "ck_tile/host/concat.hpp"#include "ck_tile/core/utility/env.hpp"#include "ck_tile/host/convolution_parameter.hpp"#include "ck_tile/ops/grouped_convolution/utils/transform_conv_fwd_to_gemm.hpp"#include "ck_tile/ops/grouped_convolution/utils/grouped_convolution_utils.hpp"Go to the source code of this file.
Classes | |
| struct | ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType > |
| The Grouped Convolution kernel device arguments. More... | |
| struct | ck_tile::GroupedConvolutionForwardKernel< GroupedConvTraitsType, TilePartitioner_, GemmPipeline_, EpiloguePipeline_ > |
| The Grouped Convolution Forward kernel template. More... | |
Namespaces | |
| ck_tile | |