| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | cross_entropy_impl.cu | 25 const size_t class_num, T *loss) { in CrossEntropyWithSparseKernel()
41 const size_t class_num, T *loss) { in LargeBatchCrossEntropyWithSparseKernel()
55 const size_t class_num, T *grad) { in CrossEntropyGradWithSparseKernel()
68 …ossEntropyKernel(const T *input0, const S *input1, const size_t batch_size, const size_t class_num, in CrossEntropyKernel()
101 …ntropyWithSparse(const T *logits, const S *labels, const size_t batch_size, const size_t class_num, in CrossEntropyWithSparse()
114 const size_t class_num, T *grad, cudaStream_t cuda_stream) { in CrossEntropyGradWithSparse()
121 …tropy(const T *logits, const S *labels, const size_t batch_size, const size_t class_num, T *losses, in CrossEntropy()
|
| D | sparse_cross_entropy_cuda_impl.cu | 21 …id CalCrossEntropyKernel(const float *logits, T *labels, const int batch_size, const int class_num, in CalCrossEntropyKernel()
40 …alCrossEntropyGradKernel(const float *logits, T *labels, const int batch_size, const int class_num, in CalCrossEntropyGradKernel()
55 …CrossEntropy(const float *logits, T *labels, const int batch_size, const int class_num, float *los… in CalCrossEntropy()
62 …sEntropyGrad(const float *logits, T *labels, const int batch_size, const int class_num, float *gra… in CalCrossEntropyGrad()
|
| D | multilabel_margin_loss_impl.cu | 24 … void PreHandleKernel(const int *target, int *is_target, const int batch_size, const int class_num, in PreHandleKernel()
46 int class_num, int64_t reduction, T *output) { in MultilabelMarginLossKernel()
106 … int class_num, int64_t reduction, T *output, T *output_tmp) { in MultilabelMarginLossSumKernel()
120 int class_num, int64_t reduction, half *output) { in MultilabelMarginLossKernel()
181 … const int batch_size, int class_num, int64_t reduction, half *output, in MultilabelMarginLossSumKernel()
196 int class_num, int64_t reduction, T *output, T *output_tmp, in CalMultilabelMarginLoss()
|
| D | multilabel_margin_loss_grad_impl.cu | 25 … const int *is_target, const int batch_size, const int class_num, in MultilabelMarginLossGradKernel()
90 … const int *is_target, const int batch_size, const int class_num, in MultilabelMarginLossGradKernel()
156 … const int batch_size, const int class_num, int64_t reduction, T *output_grad, in CalMultilabelMarginLossGrad()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/train/metrics/ |
| D | roc.py | 75 def __init__(self, class_num=None, pos_label=None): argument
114 def _roc_eval(self, y_pred, y, class_num, pos_label, sample_weights=None): argument
144 def _roc(self, y_pred, y, class_num=None, pos_label=None, sample_weights=None): argument
192 def _precision_recall_curve_update(y_pred, y, class_num, pos_label): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/hypercomplex/ |
| D | resnet.py | 489 def resnet18(class_num=10): argument
511 def resnet34(class_num=10): argument
533 def resnet50(class_num=10): argument
554 def se_resnet50(class_num=1001): argument
576 def resnet101(class_num=1001): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/ |
| D | nllloss_grad_fp32.c | 23 …const float *total_weight, float *logits_grad, int batch, int class_num, ReductionType reduction_t… in NLLLossGrad()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/model/ |
| D | resnet.py | 244 def resnet50(class_num=10): argument
264 def resnet101(class_num=1001): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/resnet50/src/ |
| D | resnet.py | 244 def resnet50(class_num=10): argument
264 def resnet101(class_num=1001): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | non_max_suppression_fp32.cc | 112 …NonMaxSuppressionCPUKernel::Run_Selecte(bool simple_out, int box_num, int batch_num, int class_num, in Run_Selecte()
233 int class_num = score_dims.at(kClassIndex); in Run() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_auto_parallel_resnet_sharding_propagation2.py | 212 def resnet50(class_num=10, matmul_stra=None, squeeze_stra=None): argument
|
| D | test_auto_parallel_resnet_sharding_propagation.py | 213 def resnet50(class_num=10, matmul_stra=None, squeeze_stra=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | non_max_suppression_fp32.c | 113 int class_num = score_dims[Index1]; in NonMaxSuppressionSelecte() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/st/python/import_ms_and_mslite/ |
| D | resnet.py | 494 def resnet50(class_num=10): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/debug/ |
| D | resnet.py | 489 def resnet50(class_num=10): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/auto_parallel/ |
| D | cell_shard.py | 239 def resnet50(class_num=10): argument
|
| D | resnet50_expand_loss.py | 240 def resnet50(class_num=10): argument
|