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.