Searched refs:smooth_l1_loss (Results 1 – 12 of 12) sorted by relevance
14 from .smooth_l1_loss import smooth_l1_loss
6 def smooth_l1_loss( function
65 gpu, mshadow_op::smooth_l1_loss>);
120 .set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Compute<cpu, mshadow_op::smooth_l1_loss>)
411 IMPLEMENT_BINARY_WORKLOAD_FWD(mxnet::op::mshadow_op::smooth_l1_loss); // NOLINT()
1087 struct smooth_l1_loss : public mxnet_op::tunable { struct
675 F<mshadow_op::smooth_l1_loss>(in, ScalarExp<DType>(sigma2)));856 struct smooth_l1_loss {