o Ti@sBddlmZmZddlmZmZddlmZeGdddZdS))ListTuple) dataclassfield)Graphc@seZdZUdZeed<dZeed<dZeed<e e dZ e e eeeefed<e e dZe e eeeefed<e e dZe e eeefed <e e dZe e eeefed <e e dZe e eefed <e e dZe e eefed <e e dZe e eee ed ffed<dS)ProfilingResultN fwd_graph bwd_graphFneeds_backward)default_factoryfwd_membwd_memfwd_timebwd_timefwd_tensor_sizesbwd_tensor_sizes. param_indices)__name__ __module__ __qualname__rr__annotations__r r boolrlistr rrstrintr rrrrrrrX/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/compile/profilers/__init__.pyr s   ""  ,rN) typingrr dataclassesrrtorch.fxrrrrrrs