microsoft / tensorflow-directml-plugin

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

Remove int->float cast for ReduceMin and ReduceMax ops #322

Closed PatriceVignola closed 1 year ago

PatriceVignola commented 1 year ago

Recent versions of DML support all types for ReduceMin and ReduceMax, so we don't need to cast to float anymore.