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

Method type

deepspeed/runtime/sparse_tensor.py:39–40  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

37
38 @staticmethod
39 def type():
40 return "deepspeed.SparseTensor"
41
42 def to_dense(self):
43 it = self.indices.unsqueeze(1)

Callers 15

init_softmax_inputsFunction · 0.80
transpose_mask_for_sparseMethod · 0.80
make_lutMethod · 0.80
make_sdd_lutMethod · 0.80
make_dxx_lutMethod · 0.80
_ref_forwardMethod · 0.80
forwardMethod · 0.80
_ref_forwardMethod · 0.80
_check_parityMethod · 0.80
funcFunction · 0.80
cuda_wf6af16_linearFunction · 0.80
preprocess_weightFunction · 0.80

Calls

no outgoing calls

Tested by 1

init_softmax_inputsFunction · 0.64