Skip to content

tensorflow-directml-plugin 0.3.0

Pre-release
Pre-release
Compare
Choose a tag to compare
@maggie1059 maggie1059 released this 13 Dec 05:17
· 7 commits to main since this release
e58bf87

The Python packages are available as a PyPI release. To download the latest python package automatically, simply pip install tensorflow-directml-plugin.

Changes in 0.3.0

  • Set tensorflow-cpu==2.10.0 as a hard dependency due to incompatibility with Keras 2.11's default optimizers.
  • Fix overflow in BatchNorm ops when float16 or mixed precision is used.
  • Remove unnecessary Cast operation in ReduceMin and ReduceMax ops.