tree 3891ecb145d8bd7aaf6538bdf52ec34b6196629c
parent 75dbe88b05e2653d0c0b7e0be651383edb33ee61
author Richard Zou <zou3519@gmail.com> 1640186477 -0800
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> 1640186644 -0800

Fix some Composite Compliance problems with binary_cross_entropy backward (#70198)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70198

This PR fixes composite compliance problems with:
- binary_cross_entropy's backward formula
- binary_cross_entropy_with_logits's backward formula
- binary_cross_entropy's double backward formula

It does so by adding checks for areAnyTensorSubclassLike.

Test Plan:
- I tested everything with functorch.
- We are going to do https://github.com/pytorch/pytorch/issues/69530 in
the future so we have a way of testing this in core. I need the
binary_cross_entropy ones for something right now and didn't want to
wait until we come up with a solution for #69530.

Reviewed By: Chillee

Differential Revision: D33246995

Pulled By: zou3519

fbshipit-source-id: 310ed3196b937d01b189870b86a6c5f77f9258b4