| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | bce_with_logits_loss_impl.cu | 58 __global__ void BCEWithLogitsLossMain(size_t size, const T *predict, const T *target, const T *shap… in BCEWithLogitsLossMain()
71 __global__ void BCEWithLogitsLossMain(size_t size, const half *predict, const half *target, in BCEWithLogitsLossMain()
92 void CalBCEWithLogitsLoss(const size_t input_size, const T *predict, const T *target, const size_t … in CalBCEWithLogitsLoss()
|
| /third_party/ffmpeg/libavcodec/ |
| D | ffv1_template.c | 23 static inline int RENAME(predict)(TYPE *src, TYPE *last) in RENAME() argument
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_bce_with_logits_loss.py | 31 def construct(self, predict, target, weight, pos_weight): argument
|
| D | test_nll_loss.py | 31 def construct(self, predict, target, weight): argument
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | soft_margin_loss_grad.cc | 32 …auto predict = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShap… in SoftMarginLossGradInferShape() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | soft_margin_loss.cc | 32 …auto predict = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape… in SoftMarginLossInferShape() local
|
| /third_party/mindspore/tests/ut/python/communication/ |
| D | test_data_parallel_lenet.py | 59 def __init__(self, predict, label, length=3): argument
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_bool_grad.py | 31 def __init__(self, predict, label, length=3): argument
|
| D | test_full_batch.py | 32 def __init__(self, predict, label, length=3): argument
|
| D | test_transpose.py | 30 def __init__(self, predict, label, length=3): argument
|
| D | test_topk.py | 27 def __init__(self, predict, label, length=3): argument
|
| D | test_gathernd.py | 27 def __init__(self, predict, label, length=3): argument
|
| D | test_model_with_loss.py | 27 def __init__(self, predict, label, length=3): argument
|
| D | test_one_dev.py | 38 def __init__(self, predict, label, length=3): argument
|
| D | test_prelu_cell.py | 35 def __init__(self, predict, label, length=3, input_num=2): argument
|
| D | test_loss_and_o2_level.py | 27 def __init__(self, predict, label, length=3): argument
|
| D | test_select.py | 27 def __init__(self, predict, label, length=3): argument
|
| D | test_bias_add.py | 40 def __init__(self, predict, label, length=3): argument
|
| D | test_parameter_merge.py | 28 def __init__(self, predict, label, length=3): argument
|
| D | test_auto_parallel_for_loop_simplify.py | 28 def __init__(self, predict, label, length=3): argument
|
| D | test_range.py | 27 def __init__(self, predict, label, length=3): argument
|
| D | test_batchnorm_batch_parallel.py | 40 def __init__(self, predict, label, length=3): argument
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_nn_ops.py | 43 def bprop(predict, label, out, dout): argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | smooth_l1_loss_grad.cc | 41 auto predict = reinterpret_cast<float *>(in_tensors_.at(0)->MutableData()); in Execute() local
|
| D | smooth_l1_loss.cc | 32 auto predict = reinterpret_cast<float *>(in_tensors_.at(0)->MutableData()); in Execute() local
|