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. |