Turn on heuristic (mostly-NHWC) convolution layout assignment for (V100, fp16) by default.
Also increase the column reduction tile size to reduce atomic operations. PiperOrigin-RevId: 198126505
Loading
Please sign in to comment
Also increase the column reduction tile size to reduce atomic operations. PiperOrigin-RevId: 198126505