MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / sum

Method sum

imperative/python/megengine/xla/rules/hlotensor.py:171–174  ·  view source on GitHub ↗
(self, axis, keepdims=False)

Source from the content-addressed store, hash-verified

169 return typecvt(self, dtype)
170
171 def sum(self, axis, keepdims=False):
172 from .reduction import sum
173
174 return sum(self, axis, keepdims)
175
176 def mean(self, axis, keepdims=False):
177 from .reduction import mean

Callers 15

checkFunction · 0.45
_multinomialFunction · 0.45
apply_opMethod · 0.45
apply_opMethod · 0.45
get_cosineFunction · 0.45
softmaxFunction · 0.45
softmax_gradFunction · 0.45
_normalize_gradFunction · 0.45
layer_norm_gradFunction · 0.45
group_norm_gradFunction · 0.45
indexing_with_tensor_indexFunction · 0.45
splitFunction · 0.45

Calls 1

sumFunction · 0.85

Tested by 15

test_advance_indexingFunction · 0.36
test_advance_indexing_with_subtensorFunction · 0.36
runFunction · 0.36
bn_trainFunction · 0.36
forwardMethod · 0.36
test_detachFunction · 0.36
_test_optimizerFunction · 0.36
train_funcFunction · 0.36
test_bn_no_track_statFunction · 0.36
test_bn_no_track_stat2Function · 0.36
test_multinomial_opFunction · 0.36
test_MultinomialRNGFunction · 0.36