This is the right behavior for TF and JAX and matches what TF does on GPU. It doesn't match TF on CPU, but that's really a TF bug. PiperOrigin-RevId: 353657779 |
||
|---|---|---|
| .. | ||
| Dialect | ||
| utils | ||
| CMakeLists.txt | ||
This is the right behavior for TF and JAX and matches what TF does on GPU. It doesn't match TF on CPU, but that's really a TF bug. PiperOrigin-RevId: 353657779 |
||
|---|---|---|
| .. | ||
| Dialect | ||
| utils | ||
| CMakeLists.txt | ||