Searched refs:softmax_cross_entropy_with_logits (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_softmax_cross_entropy_with_logits_op.py | 46 softmax_cross_entropy_with_logits = NetSoftmaxCrossEntropyWithLogits()
47 output = softmax_cross_entropy_with_logits(logits, labels)
53 softmax_cross_entropy_with_logits = NetSoftmaxCrossEntropyWithLogits()
54 output = softmax_cross_entropy_with_logits(logits, labels)
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_softmax_cross_entropy_with_logits_op.py | 46 softmax_cross_entropy_with_logits = NetSoftmaxCrossEntropyWithLogits()
47 output = softmax_cross_entropy_with_logits(logits, labels)
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_softmax_cross_entropy_with_logits.py | 47 softmax_cross_entropy_with_logits = NetSoftmaxCrossEntropyWithLogits()
48 result_open_gk = softmax_cross_entropy_with_logits(logits, labels)
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pipeline/infer/ |
| D | primitive_test.py | 36 softmax_cross_entropy_with_logits = P.SoftmaxCrossEntropyWithLogits() variable
40 return softmax_cross_entropy_with_logits(logits, labels)
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | __init__.py | 55 from .softmax_cross_entropy_with_logits import SoftmaxCrossEntropyWithLogits
|
| /third_party/mindspore/tests/vm_impl/ |
| D | vm_interface.py | 47 setattr(vm, 'softmax_cross_entropy_with_logits', softmax_cross_entropy_with_logits)
|
| D | nn_ops_vm_impl.py | 305 loss, dx = vm.softmax_cross_entropy_with_logits(logits, labels)
323 loss, dx = vm.softmax_cross_entropy_with_logits(logits, one_hot_label)
|
| D | vm_me.py | 551 def softmax_cross_entropy_with_logits(logits, labels): function
|
| /third_party/mindspore/tests/ut/python/pynative_mode/vm/ |
| D | test_vm.py | 265 loss, dx = vm.softmax_cross_entropy_with_logits(logits, labels)
|
| /third_party/mindspore/mindspore/ops/_op_impl/tbe/ |
| D | __init__.py | 91 from .softmax_cross_entropy_with_logits import _softmax_cross_entropy_with_logits_tbe
|
| /third_party/mindspore/config/ |
| D | op_info.config | 222 …lse, "binfile_name": "softmax_cross_entropy_with_logits.so", "compute_cost": 10, "kernel_name": "s…
|