d1828625ab
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 |