|
mxnet
|
|
Functions | |
| dnnl_status_t DNNL_API | dnnl_get_default_fpmath_mode (dnnl_fpmath_mode_t *mode) |
| dnnl_status_t DNNL_API | dnnl_set_default_fpmath_mode (dnnl_fpmath_mode_t mode) |
| dnnl_status_t DNNL_API dnnl_get_default_fpmath_mode | ( | dnnl_fpmath_mode_t * | mode | ) |
Returns the floating-point math mode that will be used by default for all subsequently created primitives.
| mode | Output FP math mode. |
| dnnl_status_t DNNL_API dnnl_set_default_fpmath_mode | ( | dnnl_fpmath_mode_t | mode | ) |
Sets the floating-point math mode that will be used by default for all subsequently created primitives.
| mode | FP math mode. The possible values are: dnnl_fpmath_mode_strict, dnnl_fpmath_mode_bf16, dnnl_fpmath_mode_f16, dnnl_fpmath_mode_any. |
1.8.13