Searched refs:mp_diff_buffer_and_reduce (Results 1 – 1 of 1) sorted by relevance
95 mp_diff_buffer_and_reduce = MixedPrecision( variable100 mp_configs.extend([mp_diff_buffer_and_reduce])134 str(mp_diff_buffer_and_reduce): "mp_diff_buffer_reduce",557 mp = default_mp if not nccl_supports_bf16 else mp_diff_buffer_and_reduce1059 mp = default_mp if not nccl_supports_bf16 else mp_diff_buffer_and_reduce1072 mp = default_mp if not nccl_supports_bf16 else mp_diff_buffer_and_reduce