MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed / unflatten_bench.py

File unflatten_bench.py

tests/benchmarks/unflatten_bench.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1# Copyright (c) Microsoft Corporation.
2# SPDX-License-Identifier: Apache-2.0
3
4# DeepSpeed Team

Callers

nothing calls this directly

Calls 12

UtilsBuilderClass · 0.90
get_acceleratorFunction · 0.90
_flatten_dense_tensorsFunction · 0.90
_unflatten_dense_tensorsFunction · 0.90
flattenFunction · 0.85
line_profilemeFunction · 0.70
cprofilemeFunction · 0.70
timemeFunction · 0.70
loadMethod · 0.45
manual_seedMethod · 0.45
toMethod · 0.45
device_nameMethod · 0.45

Tested by

no test coverage detected