Searched refs:logsoftmax (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_hybird_parallel_activation.py | 140 self.logsoftmax = P.LogSoftmax().shard(strategy3)
143 out = self.logsoftmax(self.matmul1(x, y))
169 self.logsoftmax = P.LogSoftmax().shard(strategy3)
173 out = self.logsoftmax(self.softmax(self.matmul2(out, b)))
198 self.logsoftmax = P.LogSoftmax().shard(strategy6)
202 out = self.logsoftmax(self.softmax(self.matmul2(out, b)))
230 self.logsoftmax = P.LogSoftmax().shard(strategy6)
234 out = self.logsoftmax(self.softmax(self.matmul2(out, b)))
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_logsoftmax.py | 29 self.logsoftmax = P.LogSoftmax(axis)
32 return self.logsoftmax(x)
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_logsoftmax_op.py | 46 self.logsoftmax = P.LogSoftmax(axis)
49 return self.logsoftmax(x)
|
| /third_party/mindspore/tests/ut/python/nn/ |
| D | test_activation.py | 52 self.logsoftmax = nn.LogSoftmax(dim)
55 return self.logsoftmax(x)
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_logsoftmax_op.py | 48 self.logsoftmax = P.LogSoftmax(axis)
51 return self.logsoftmax(x)
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | __init__.py | 40 from .logsoftmax import LogSoftmax
|
| /third_party/mindspore/tests/vm_impl/ |
| D | vm_interface.py | 53 setattr(vm, 'logsoftmax', logsoftmax)
|
| D | nn_ops_vm_impl.py | 79 return Tensor(vm.logsoftmax(x))
|
| D | vm_me.py | 630 def logsoftmax(x): function
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | nn_norm_ops_declare.cc | 88 OUTPUT_MAP(LogSoftmaxV2) = {{0, OUTPUT_DESC(logsoftmax)}};
|
| /third_party/mindspore/mindspore/nn/loss/ |
| D | loss.py | 1334 self.logsoftmax = nn.LogSoftmax(1)
1351 log_probability = self.logsoftmax(logits)
|
| /third_party/mindspore/mindspore/ops/_op_impl/tbe/ |
| D | __init__.py | 186 from .logsoftmax import _logsoftmax_tbe
|
| /third_party/mindspore/config/ |
| D | op_info.config | 311 …type": "required", "shape": "all"}], "outputs": [{"index": 0, "name": "logsoftmax", "need_compile"…
|