LOGSIGMOID_dt_bfloat16.cu 245 字节
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(LOGSIGMOID, cb)
#define KERN_IMPL_ARITY    1
#define KERN_IMPL_CTYPE    dt_bfloat16
#include "../kern_impl.inl"
#endif