Skip to content
Branch: master
Find file History
lindong28 and tensorflower-gardener Currently there is only one implementation of the thread pool (i.e. t…
…ensorflow/core/lib/core/threadpool.cc) for CPU threads in TensorFlow. In order to make it easy to innovate, develop and test other thread scheduling algorithm, we would like to support multiple thread pool implementation and allow user to choose the thread pool type in the config.

This patch made the following changes:

- Added interface class tensorflow::thread::ThreadPoolInterface
- Added APIs that allow user to pass inter-op threadpool via RunCallable()

PiperOrigin-RevId: 253401360
Latest commit 1d8d2a0 Jun 16, 2019
You can’t perform that action at this time.