forked from jiuyuan/InfiniTensor
parent
5b6698bac7
commit
71ca4459d9
|
@ -2,6 +2,7 @@
|
|||
#include "core/kernel.h"
|
||||
#include "cuda/cuda_kernel_wihtout_config.h"
|
||||
#include "cuda/cuda_runtime.h"
|
||||
|
||||
namespace infini {
|
||||
class BatchNormCudnn : public CudaKernelWithoutConfig {
|
||||
void compute(const Operator &_op,
|
||||
|
|
Loading…
Reference in New Issue