OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 042c7fd10a
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '042c7fd10a'
${ noResults }
MegEngine/dnn/src/cuda/elemwise/kimpl/LOGSIGMOID_dt_bfloat16.cu

8 lines
245 B
Raw Blame History 

  1. // generated by gen_elemwise_kern_impls.py

  2. #if !MEGDNN_DISABLE_FLOAT16

  3. #define KERN_IMPL_MODE(cb) MEGDNN_ELEMWISE_MODE_ENABLE(LOGSIGMOID, cb)

  4. #define KERN_IMPL_ARITY    1

  5. #define KERN_IMPL_CTYPE    dt_bfloat16

  6. #include "../kern_impl.inl"

  7. #endif