MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed / ds_shape

Function ds_shape

deepspeed/utils/debug.py:61–65  ·  view source on GitHub ↗
(param)

Source from the content-addressed store, hash-verified

59
60
61def ds_shape(param):
62 if hasattr(param, "ds_shape"):
63 return param.ds_shape
64 else:
65 return param.shape
66
67
68def debug_param2name_id(param):

Callers 3

debug_param2name_id_shapeFunction · 0.85
debug_param2name_id_shape_deviceFunction · 0.85
debug_param2name_id_shape_statusFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected