Add a small int64 workaround for elementwise ops that don't support it

microsoft / tensorflow-directml

Fork of TensorFlow accelerated by DirectML

Apache License 2.0

454 stars 32 forks source link

Add a small int64 workaround for elementwise ops that don't support it #391

Closed PatriceVignola closed 1 year ago

PatriceVignola commented 1 year ago

It turns out that DIVIDE, MODULUS_FLOOR, DIFFERENCE_SQUARE, MODULUS_TRUNCATE and POW don't yet support emulated int64 in DirectML, so we can't entirely get rid of int64 workarounds.