Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
convolution_backward_data_specialization.hpp 332 Bytes
Newer Older
ltqin's avatar
NHWC conv 2d: bwd fp32/fp16/bfp16/int8, Device level tuning and host API (#92)  
ltqin committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 
#ifndef CONVOLUTION_BACKWARD_DATA_SPECIALIZATION
#define CONVOLUTION_BACKWARD_DATA_SPECIALIZATION

namespace ck {
namespace tensor_operation {
namespace device {

enum ConvolutionBackwardDataSpecialization_t
{
    Default,
    Filter1x1Stride1Pad0,
};

} // namespace device
} // namespace tensor_operation
} // namespace ck
#endif