DeepSpeed
Generalize the model input format of the flops profiler get_model_profile() API
#1768
Merged
Merged