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

Function GetWorkersCount

tensorflow/core/kernels/meta_support.cc:89–94  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

87}
88
89int GetWorkersCount(OpKernelContext* tf_context) {
90 if (g_num_threads == 0) {
91 return tf_context->device()->tensorflow_cpu_worker_threads()->num_threads;
92 }
93 return g_num_threads;
94}
95
96typedef gemmlowp::meta::SimpleContext<gemmlowp::WorkersPool> LocalContext;
97

Callers 2

QuantizedGemmImplFunction · 0.85
MultiThreadTransform1DFunction · 0.85

Calls 2

deviceMethod · 0.45

Tested by

no test coverage detected