Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsmooth_l1_loss.h35 …SmoothL1Loss() : BaseOperator(kNameSmoothL1Loss) { InitIOName({"prediction", "target"}, {"output"}… in SmoothL1Loss() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dsmooth_l1_loss_impl.cu122 cudaError_t SmoothL1Loss(const SmoothL1LossReductionMode mode, const int64_t input_size, const floa… in SmoothL1Loss() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/loss/
Dloss.py545 class SmoothL1Loss(LossBase): class
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dnn_ops.py2355 class SmoothL1Loss(Primitive): class