Add lazy shape inference for logical boolean ops (#78004)
Add lazy shape inference for logical boolean ops
- logical_and
- logical_not
- logical_or
- logical_xor
Uses helper functions defined at https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/ExpandUtils.h#L21 to infer the shape.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78004
Approved by: https://github.com/wconstab