| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| 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 *logits, const S *labels, const size_t batch_size, const size_t class_num, in CrossEntropyKernel()
82 …hSparse(const T *logits, const S *labels, const size_t batch_size, const size_t class_num, T *loss, in CrossEntropyWithSparse()
93 …pyGradWithSparse(const T *logits, const S *labels, const size_t batch_size, const size_t class_num, 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 …id CalCrossEntropyKernel(const float *logits, T *labels, const int batch_size, const int class_num, in CalCrossEntropyKernel()
41 …alCrossEntropyGradKernel(const float *logits, T *labels, const int batch_size, const int class_num, 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()
|
| /third_party/mindspore/mindspore/nn/metrics/ |
| D | roc.py | 71 def __init__(self, class_num=None, pos_label=None): argument
84 def _precision_recall_curve_update(self, y_pred, y, class_num, pos_label): argument
138 def _roc_eval(self, y_pred, y, class_num, pos_label, sample_weights=None): argument
167 def roc(self, y_pred, y, class_num=None, pos_label=None, sample_weights=None): argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | resnet.py | 335 def resnet50(class_num=10): argument
355 def se_resnet50(class_num=1001): argument
376 def resnet101(class_num=1001): argument
|
| /third_party/mindspore/tests/st/ps/multi_full_ps/ |
| D | resnet.py | 244 def resnet50(class_num=10): argument
265 def resnet101(class_num=1001): argument
|
| /third_party/mindspore/tests/ut/python/model/ |
| D | resnet.py | 244 def resnet50(class_num=10): argument
264 def resnet101(class_num=1001): argument
|
| /third_party/mindspore/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/tests/st/quantization/resnet50_quant/ |
| D | resnet_quant_manual.py | 307 def resnet50_quant(class_num=10): argument
328 def resnet101_quant(class_num=1001): argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/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/tests/ut/python/communication/ |
| D | test_data_parallel_resnet.py | 297 class_num = 10 variable
|
| /third_party/mindspore/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
|
| D | test_auto_parallel_resnet.py | 214 def resnet50(class_num=10): argument
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | resnet50_expand_loss.py | 239 def resnet50(class_num=10): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | yolov3.py | 251 def resnet18(class_num=10): argument
|