transform_conv_bwd_data_to_gemm_v1.hpp File Reference

• include
• ck
• tensor_operation
• operator_transform

Classes | Namespaces | Macros

transform_conv_bwd_data_to_gemm_v1.hpp File Reference

#include "ck/utility/common_header.hpp"
#include "ck/tensor_description/tensor_descriptor.hpp"
#include "ck/tensor_description/tensor_descriptor_helper.hpp"
#include "ck/tensor_operation/gpu/device/tensor_layout.hpp"
#include "ck/tensor_operation/gpu/device/convolution_backward_data_specialization.hpp"
#include "ck/tensor_operation/gpu/device/matrix_padder.hpp"

Go to the source code of this file.

Classes
struct	ck::tensor_operation::TransformConvBwdDataToGemm_v1< NDimSpatial, ConvBwdDataSpecialization, AK1, BK1, GemmMPerBlock, GemmNPerBlock, GemmKPerBlock, DoPadGemmM, DoPadGemmN, ALayout, BLayout, CLayout, SplitN, ADataType, CDataType, NumGroupsToMerge, IndexType, CTranspose >

Namespaces
namespace	ck
namespace	ck::tensor_operation

Macros
#define	CK_USE_CUSTOM_TENSOR_TRANSFORM_FOR_BWD_DATA_OUT   1
	Enable custom tensor transform for convolution backward data output.

Macro Definition Documentation

◆ CK_USE_CUSTOM_TENSOR_TRANSFORM_FOR_BWD_DATA_OUT

#define CK_USE_CUSTOM_TENSOR_TRANSFORM_FOR_BWD_DATA_OUT   1

Enable custom tensor transform for convolution backward data output.

When set to 1, this macro enables a custom transformation of the output tensor in convolution backward data operations.