Home
last modified time | relevance | path

Searched refs:TensorflowGemmContext (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmeta_support.cc156 TensorflowGemmContext context(workers.num_threads, workers.workers); in QuantizedGemmImpl()
157 MultiThreadGemm<TensorflowGemmContext, Params>(&context, params); in QuantizedGemmImpl()
169 TensorflowGemmContext context(workers.num_threads, workers.workers); in MultiThreadTransform1D()
170 gemmlowp::meta::MultiThreadTransform1D<TensorflowGemmContext, Params, in MultiThreadTransform1D()
Dquantization_utils.h942 class TensorflowGemmContext : public gemmlowp::MultiThreadGemmContextBase {
944 TensorflowGemmContext(int num_threads, thread::ThreadPool* workers) in TensorflowGemmContext() function
954 TF_DISALLOW_COPY_AND_ASSIGN(TensorflowGemmContext);
Dquantized_matmul_op.cc56 TensorflowGemmContext context(worker_threads.num_threads, in GemmlowpMultiply()
Dquantized_conv_ops.cc428 TensorflowGemmContext context(worker_threads.num_threads, in operator ()()