Home
last modified time | relevance | path

Searched defs:SmoothL1Loss (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dsmooth_l1_loss.h33SmoothL1Loss() : PrimitiveC(kNameSmoothL1Loss) { InitIOName({"prediction", "target"}, {"output"});… in SmoothL1Loss() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dsmooth_l1_loss_impl.cu34 void SmoothL1Loss(const int &input_size, const float &beta, const T *prediction, const T *target, T… in SmoothL1Loss() function
/third_party/mindspore/mindspore/nn/loss/
Dloss.py385 class SmoothL1Loss(LossBase): class
/third_party/mindspore/mindspore/ops/operations/
Dnn_ops.py2681 class SmoothL1Loss(PrimitiveWithInfer): class