PRELU_GRAD_dt_float16.cu 244 字节
Newer Older
1 2 3 4 5 6 7
// generated by gen_elemwise_kern_impls.py
#if !MEGDNN_DISABLE_FLOAT16
#define KERN_IMPL_MODE(cb) MEGDNN_ELEMWISE_MODE_ENABLE(PRELU_GRAD, cb)
#define KERN_IMPL_ARITY    3
#define KERN_IMPL_CTYPE    dt_float16
#include "../kern_impl.inl"
#endif