Searched refs:tf_context (Results 1 – 1 of 1) sorted by relevance
89 int GetWorkersCount(OpKernelContext* tf_context) { in GetWorkersCount() argument91 return tf_context->device()->tensorflow_cpu_worker_threads()->num_threads; in GetWorkersCount()118 void QuantizedGemmImpl(OpKernelContext* tf_context, const quint8* a_data, in QuantizedGemmImpl() argument136 params.scratch = CHECK_NOTNULL(GetScratch(tf_context)); in QuantizedGemmImpl()152 LocalContext local_context(GetWorkersCount(tf_context), GetWorkersPool()); in QuantizedGemmImpl()155 auto& workers = *(tf_context->device()->tensorflow_cpu_worker_threads()); in QuantizedGemmImpl()162 void MultiThreadTransform1D(OpKernelContext* tf_context, const Params& params) { in MultiThreadTransform1D() argument164 LocalContext local_context(GetWorkersCount(tf_context), GetWorkersPool()); in MultiThreadTransform1D()168 auto& workers = *(tf_context->device()->tensorflow_cpu_worker_threads()); in MultiThreadTransform1D()220 void QuantizedGemm(OpKernelContext* tf_context, bool transpose_a, in QuantizedGemm() argument[all …]