MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / SetThreadPool

Method SetThreadPool

tensorflow/core/common_runtime/memory_planner.cc:142–146  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

140}
141
142void MemoryPlanner::SetThreadPool(thread::ThreadPool* thread_pool) {
143 if (thread_pool_ == nullptr) {
144 thread_pool_ = thread_pool;
145 }
146}
147void MemoryPlanner::Schedule(std::function<void()> f) {
148 if (thread_pool_ == nullptr) {
149 f();

Callers 3

DirectSessionMethod · 0.45
TESTFunction · 0.45
RegisterGraphAsyncMethod · 0.45

Calls

no outgoing calls

Tested by 1

TESTFunction · 0.36