Searched defs:losses (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | sparse_softmax_cross_entropy_with_logits.cc | 33 …eSoftmaxCrossEntropyWithLogitsCPUKernel::ForwardPostExecute(const int *labels, const float *losses, in ForwardPostExecute()
54 …arseSoftmaxCrossEntropyWithLogitsCPUKernel::GradPostExecute(const int *labels, const float *losses, in GradPostExecute()
91 float *losses = static_cast<float *>(workspace()); in Execute() local
136 float *losses = static_cast<float *>(workspace()); in Run() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | sparse_softmax_cross_entropy_with_logits_cpu_kernel.cc | 72 …eSoftmaxCrossEntropyWithLogitsCPUKernel::ForwardPostExecute(const int *labels, const float *losses, in ForwardPostExecute()
89 …arseSoftmaxCrossEntropyWithLogitsCPUKernel::GradPostExecute(const int *labels, const float *losses, in GradPostExecute()
133 const auto *losses = reinterpret_cast<float *>(workspace[0]->addr); in Launch() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | cross_entropy_impl.cu | 69 T epsilon, T *losses, T *dlogits) { in CrossEntropyKernel()
100 …tropy(const T *logits, const S *labels, const size_t batch_size, const size_t class_num, T *losses, in CrossEntropy()
|
| /third_party/mindspore/tests/st/ps/multi_full_ps/ |
| D | test_multi_full_ps.py | 108 losses = [] variable
|
| /third_party/mindspore/tests/st/fl/cross_silo_lenet/ |
| D | test_cross_silo_lenet.py | 110 losses = [] variable
|
| /third_party/mindspore/tests/st/fl/mobile/ |
| D | test_mobile_lenet.py | 136 losses = [] variable
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/ |
| D | test_hybrid_train_lenet.py | 156 losses = [] variable
|