| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | cross_entropy_impl.cu | 24 __global__ void CrossEntropyWithSparseKernel(const T *logits, const S *labels, const size_t batch_s… in CrossEntropyWithSparseKernel()
40 __global__ void LargeBatchCrossEntropyWithSparseKernel(const T *logits, const S *labels, const size… in LargeBatchCrossEntropyWithSparseKernel()
54 __global__ void CrossEntropyGradWithSparseKernel(const T *logits, const S *labels, const size_t bat… in CrossEntropyGradWithSparseKernel()
68 __global__ void CrossEntropyKernel(const T *logits, const S *labels, const size_t batch_size, const… in CrossEntropyKernel()
82 void CrossEntropyWithSparse(const T *logits, const S *labels, const size_t batch_size, const size_t… in CrossEntropyWithSparse()
93 void CrossEntropyGradWithSparse(const T *logits, const S *labels, const size_t batch_size, const si… in CrossEntropyGradWithSparse()
100 void CrossEntropy(const T *logits, const S *labels, const size_t batch_size, const size_t class_num… in CrossEntropy()
|
| D | sparse_cross_entropy_cuda_impl.cu | 22 __global__ void CalCrossEntropyKernel(const float *logits, T *labels, const int batch_size, const i… in CalCrossEntropyKernel()
41 __global__ void CalCrossEntropyGradKernel(const float *logits, T *labels, const int batch_size, con… in CalCrossEntropyGradKernel()
56 void CalCrossEntropy(const float *logits, T *labels, const int batch_size, const int class_num, flo… in CalCrossEntropy()
63 void CalCrossEntropyGrad(const float *logits, T *labels, const int batch_size, const int class_num,… in CalCrossEntropyGrad()
|
| D | sigmoid_cross_entropy_with_logits_impl.cu | 20 __global__ void SigmoidCrossEntropyWithLogitsKernel(const size_t size, const T *logits, const S *la… in SigmoidCrossEntropyWithLogitsKernel()
29 void SigmoidCrossEntropyWithLogits(const size_t size, const T *logits, const S *labels, T *outputs, in SigmoidCrossEntropyWithLogits()
|
| D | sigmoid_cross_entropy_with_logits_grad_impl.cu | 20 __global__ void SigmoidCrossEntropyWithLogitsGradKernel(const size_t size, const T *logits, const S… in SigmoidCrossEntropyWithLogitsGradKernel()
33 void SigmoidCrossEntropyWithLogitsGrad(const size_t size, const T *logits, const S *labels, const T… in SigmoidCrossEntropyWithLogitsGrad()
|
| /third_party/mindspore/mindspore/nn/loss/ |
| D | loss.py | 103 def construct(self, logits, labels): argument
117 def construct(self, logits, labels): argument
192 def construct(self, logits, labels): argument
257 def construct(self, logits, labels): argument
309 def construct(self, logits, label): argument
378 def construct(self, logits, label): argument
444 def construct(self, logits, labels): argument
493 def construct(self, logits, labels): argument
574 def construct(self, logits, labels): argument
636 def construct(self, logits, label): argument
[all …]
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | assessment_method.py | 28 def update(self, logits, labels): argument
48 def update(self, logits, labels): argument
70 def update(self, logits, labels): argument
96 def update(self, logits, labels): argument
123 def update(self, logits, labels): argument
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | loss.py | 94 def construct(self, logits, label, input_mask): argument
127 def _check_input(self, logits, label, input_mask): argument
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_sigmoid_cross_entropy_with_logits.py | 31 def construct(self, logits, labels): argument
40 def construct(self, logits, labels, dout): argument
|
| D | test_softmax_cross_entropy_with_logits.py | 30 def construct(self, logits, labels): argument
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pipeline/infer/ |
| D | primitive_test.py | 39 def get_softmax_cross_entropy_with_logits(logits, labels): argument
50 def __call__(self, logits, labels): argument
64 def get_tensor_to_scalar(logits, labels): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | softmax_cross_entropy_with_logits_cpu_kernel.cc | 66 void SoftmaxCrossEntropyWithLogitsCPUKernel::ForwardPostExecute(const float *logits, const float *l… in ForwardPostExecute()
101 const auto *logits = reinterpret_cast<float *>(workspace[0]->addr); in Launch() local
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_softmax_cross_entropy_with_logits_op.py | 29 def construct(self, logits, labels): argument
|
| D | test_sparse_softmax_cross_entropy_with_logits_op.py | 28 def construct(self, logits, labels): argument
|
| D | test_sigmoid_cross_entropy_with_logits_op.py | 30 def construct(self, logits, labels): argument
|
| D | test_sigmoid_cross_entropy_with_logits_grad_op.py | 30 def construct(self, logits, labels, dout): argument
|
| D | test_random_categorical_op.py | 32 def construct(self, logits): argument
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_softmax_cross_entropy_with_logits_op.py | 29 def construct(self, logits, labels): argument
|
| D | test_sigmoid_cross_entropy_with_logits_op.py | 30 def construct(self, logits, labels): argument
|
| D | test_sigmoid_cross_entropy_with_logits_grad_op.py | 30 def construct(self, logits, labels, dout): argument
|
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_random_categorical.py | 29 def construct(self, logits, seed=0): argument
|
| /third_party/mindspore/tests/st/networks/models/deeplabv3/src/ |
| D | losses.py | 44 def construct(self, logits, labels): argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | sigmoid_cross_entropy_with_logits.cc | 39 auto logits = reinterpret_cast<float *>(in_tensors_.at(0)->MutableData()); in Execute() local
|
| D | sigmoid_cross_entropy_with_logits_grad.cc | 40 auto logits = reinterpret_cast<float *>(in_tensors_.at(0)->MutableData()); in Execute() local
|
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | CRF.py | 75 def log_sum_exp(self, logits): argument
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | test_resnet_cifar_1p.py | 95 def construct(self, logits, label): argument
|