transform_conv_bwd_weight_to_gemm_v2.hpp File Reference#
transform_conv_bwd_weight_to_gemm_v2.hpp File Reference
#include "ck/library/utility/numeric.hpp"#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_weight_specialization.hpp"Go to the source code of this file.
Classes | |
| struct | ck::tensor_operation::TransformConvBwdWeightToGemmV2< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, NumGroupsToMerge, ConvBackwardWeightSpecialization > |
| Transform conv bwd weight to gemm v2. More... | |
Namespaces | |
| namespace | ck |
| namespace | ck::tensor_operation |