[NVIDIA TF] Prefer NHWC in layout optimizer for bfloat16 on Ampere+ GPUs.
LGTM. Thanks.
On Ampere+ GPUs bfloat16 convolutions perform better in NHWC than NCHW (similar to fp16 for Volta+ GPUs).
This PR updates the Grappler layout optimizer to take this into consideration.