AutoMixedPrecisionLists¶
- class paddle.fluid.contrib.mixed_precision.fp16_lists. AutoMixedPrecisionLists ( custom_white_list=None, custom_black_list=None, custom_black_varnames=None ) [source]
-
AutoMixedPrecisionLists is a class for black/white list. It can update pre-defined black list and white list according to users’ custom black white lists. The lists are used for an algorithm which determines op’s execution mode (fp32 or fp16).
- Parameters
-
custom_white_list (set) – Users’ custom white list.
custom_black_list (set) – Users’ custom black list.
custom_black_varnames (set) – Users’ custom black varibles’ names.