Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused.cc309 ResizeTaskParameters<T1> task_params; in operator ()() local
310 task_params.input_depth = input_depth; in operator ()()
311 task_params.top_padding = top_padding; in operator ()()
312 task_params.pad_offset = pad_offset; in operator ()()
313 task_params.resized_height = resized_height; in operator ()()
314 task_params.st = st; in operator ()()
315 task_params.left_padding = left_padding; in operator ()()
316 task_params.resized_width = resized_width; in operator ()()
317 task_params.padded_width = padded_width; in operator ()()
318 task_params.padded_height = padded_height; in operator ()()
[all …]
/external/gemmlowp/meta/
Dmulti_thread_transform.h31 std::vector<Params>* task_params) { in PrepareTransform1DTasks() argument
49 task_params->push_back(params); in PrepareTransform1DTasks()
50 Params& task = task_params->back(); in PrepareTransform1DTasks()
55 task_params->push_back(params); in PrepareTransform1DTasks()
56 Params& task = task_params->back(); in PrepareTransform1DTasks()
80 std::vector<Params> task_params; in MultiThreadTransform1D() local
82 context, params, kernel_size, &task_params)) { in MultiThreadTransform1D()
89 for (auto& task_param : task_params) { in MultiThreadTransform1D()
Dmulti_thread_gemm.h59 std::vector<Params>* task_params) { in PrepareGemmTasks() argument
82 0, params.n, task_params); in PrepareGemmTasks()
87 params.n, task_params); in PrepareGemmTasks()
93 i * n_chunk, n_chunk, task_params); in PrepareGemmTasks()
98 params.n - sum_n, task_params); in PrepareGemmTasks()
126 std::vector<Params> task_params; in MultiThreadGemm() local
128 context, params, kernel_m, kernel_n, kernel_k, &task_params)) { in MultiThreadGemm()
135 for (auto& task_param : task_params) { in MultiThreadGemm()
Dsingle_thread_gemm.h418 const Params& params, Params* task_params) { in UpdateCacheFriendlyTask() argument
419 task_params->m = m; in UpdateCacheFriendlyTask()
420 task_params->lhs = in UpdateCacheFriendlyTask()
424 task_params->n = n; in UpdateCacheFriendlyTask()
425 task_params->rhs = in UpdateCacheFriendlyTask()
429 task_params->result = in UpdateCacheFriendlyTask()
472 P task_params = params; in ExecuteDispatch3D() local
476 &task_params); in ExecuteDispatch3D()
477 Gemm<GemmExecutorPackRHS, P, m, n, k>(task_params); in ExecuteDispatch3D()
481 params, &task_params); in ExecuteDispatch3D()
[all …]