microsoft / tensorflow-directml-plugin

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

Migrate the implementation of emulated random ops to eager mode #255

Closed PatriceVignola closed 2 years ago

PatriceVignola commented 2 years ago

When dealing with a single operator, eager execution is faster because TensorFlow doesn't need to build a graph and run optimization passes on it. Since we're running a single operator within another operator, eager execution feels like the most performant option.