Build of tensorflow-directml built on October 3, 2022
The Python packages are available as a PyPI release. To download the latest python package automatically, simply pip install tensorflow-directml
.
Changes in 1.15.8
- Prevent unbounded growth of command allocator memory.
- Add unsorted segment ops for DML (emulated on the CPU).
- Add emulated support for int64.
- Add CPU emulated versions of
UnsortedSegmentSum
,UnsortedSegmentMax
,UnsortedSegmentMin
andUnsortedSegmentProd
. - Pin protobuf version between 3.61 (inclusive) and 4.0.0 (exclusive) since version >= 4.0.0 is not compatible with TensorFlow 1.15.
- Optimize output allocation for inputs that can be executed in-place and directly forwarded to the output.
- Add a DirectML kernel for
InTopKV2
.