search

microsoft / tensorflow-directml-plugin

DirectML PluggableDevice plugin for TensorFlow 2
Apache License 2.0
185 stars 25 forks source link

Remove int64 -> uint64 reinterpret in Reduce #257

Closed PatriceVignola closed 2 years ago

PatriceVignola commented 2 years ago

Now that int64 is completely supported, we don't need the int64 -> uint64 hack for ArgMin/ArgMax anymore.