Searched refs:_wrap_custom_loss_fn (Results 1 – 1 of 1) sorted by relevance
325 loss_fn = _wrap_custom_loss_fn(loss_fn) if loss_fn else None445 loss_fn=_wrap_custom_loss_fn(loss_fn) if loss_fn else None)1843 def _wrap_custom_loss_fn(loss_fn): function