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,UnsortedSegmentMinandUnsortedSegmentProd. - 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.