DeepSpeed

2151c787 - Generalize the model input format of the flops profiler get_model_profile() API (#1768)

Generalize the model input format of the flops profiler get_model_profile() API (#1768) * generalize profiler model input format * fix timining of module forward Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Jeff Rasley <jerasley@microsoft.com>