Files
composable_kernel/src/include/direct_convolution.cuh
2018-10-22 11:51:10 -05:00

13 lines
444 B
Plaintext

#pragma once
#include "device_tensor.cuh"
template <class TFloat, int NBlockDim>
__global__ void direct_convolution(DeviceTensorDescriptor in_desc,
TFloat* const in,
DeviceTensorDescriptor wei_desc,
TFloat* const wei,
DeviceTensorDescriptor out_desc,
TFloat* out)
{
}