Skip to content

tensorflow-directml-plugin 0.4.0

Pre-release
Pre-release
Compare
Choose a tag to compare
@maggie1059 maggie1059 released this 03 Feb 00:57
v0.4.0
5973109
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

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.4.0

  • Add DirectML kernels for CudnnRNNCanonicalToParams and CudnnRNNParamsToCanonical
  • Add support for grouped convolution in Conv2DBackpropFilter and Conv3DBackpropFilter
  • Add float16 support for _FusedConv2D
Assets 3
Loading