Searched refs:log_probability (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/nn/loss/ |
| D | loss.py | 1351 log_probability = self.logsoftmax(logits)
1354 log_probability = self.gather_d(log_probability, 1, self.cast(labelss, mindspore.int32))
1355 log_probability = self.squeeze(log_probability)
1357 probability = F.exp(log_probability)
1365 log_probability = log_probability * convert_weight
1369 loss = (-1 * weight * log_probability).mean(axis=1)
1371 loss = (-1 * weight * labelss * log_probability).mean(axis=-1)
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | ctc_ops_declare.cc | 38 {3, OUTPUT_DESC(log_probability)}};
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 508 auto log_probability = in InferImplCTCGreedyDecoder() local
512 AbstractBasePtrList elements = {decoded_indices, decoded_values, decoded_shape, log_probability}; in InferImplCTCGreedyDecoder()
|
| /third_party/mindspore/config/ |
| D | op_info.config | 64 …": "decoded_shape", "param_type": "required"}, {"index": 3, "name": "log_probability", "param_type…
|