xla

6c2fc734 - test backward() calls don't hit the CPU fallback (#3092)

test backward() calls don't hit the CPU fallback (#3092) * test backward() calls don't hit the CPU fallback * lint * fix requires_grad case * generalize backward helper for multiple inputs