microsoft / tensorflow-directml-plugin

DirectML PluggableDevice plugin for TensorFlow 2
Apache License 2.0
179 stars 23 forks source link

Add int32 CPU implementation for Pack #281

Closed PatriceVignola closed 1 year ago

PatriceVignola commented 1 year ago

Pack unfortunately still doesn't have a DEVICE_DEFAULT registration in TensorFlow Core, so we're force to register it here ourselves.