The maximum supported target rank of 5 is sufficient for all operations but `select`. Make the maximum target rank configurable in the rank specialization. This reduces the number of generated kernels for operations that don't require it. PiperOrigin-RevId: 376822496 |
||
|---|---|---|
| .. | ||
| mhlo | ||
| CMakeLists.txt | ||