Searched refs:exp_sum (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | softmax_int8.c | 45 int32_t exp_sum = 0; in SoftmaxInt8() local
53 exp_sum = exp_sum + Rescale(exp_val, 0, 12); in SoftmaxInt8()
55 sum_data[c] = exp_sum; in SoftmaxInt8()
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | loss.py | 103 exp_sum = self.sum(logit_exp, -1)
104 exp_sum = P.Reshape()(exp_sum, (F.shape(exp_sum)[0], 1))
105 softmax_result = self.div(logit_exp, exp_sum)
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | soft_entropy_loss_expand_parallel.py | 161 exp_sum = self.reduce_sum(exp, -1)
162 softmax_result = self.div(exp, exp_sum)
|
| D | resnet50_expand_loss.py | 272 exp_sum = self.sum(exp, -1)
273 softmax_result = self.div(exp, exp_sum)
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_auto_parallel_resnet_sharding_propagation2.py | 247 exp_sum = self.sum(exp, -1)
248 softmax_result = self.div(exp, exp_sum)
|
| D | test_auto_parallel_resnet_sharding_propagation.py | 248 exp_sum = self.sum(exp, -1)
249 softmax_result = self.div(exp, exp_sum)
|
| D | test_auto_parallel_resnet.py | 247 exp_sum = self.sum(exp, -1)
248 softmax_result = self.div(exp, exp_sum)
|