cerebras.modelzoo.data_preparation.utils.get_output_type_shapes#

cerebras.modelzoo.data_preparation.utils.get_output_type_shapes(max_seq_length, max_predictions_per_seq, mlm_only=False)[source]#