Home
last modified time | relevance | path

Searched refs:binary_cross_entropy_backward (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/functorch/
DBatchRulesLoss.cpp133 return at::binary_cross_entropy_backward(grad, input, target, weight_opt, reduction); in binary_cross_entropy_backward_plumbing()
155 grad_input = at::binary_cross_entropy_backward( in binary_cross_entropy_backward_plumbing()
160 grad_input = at::binary_cross_entropy_backward( in binary_cross_entropy_backward_plumbing()
/external/pytorch/torch/_decomp/
D__init__.py276 aten.binary_cross_entropy_backward,
Ddecompositions.py662 @register_decomposition(aten.binary_cross_entropy_backward)
665 def binary_cross_entropy_backward( function