Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc413 const size_t work_unit_size = size_A + size_B + size_C; in Compute() local
419 work_unit_size / worker_threads.num_threads; in Compute()
437 : (target_working_set_size + work_unit_size - 1) / work_unit_size; in Compute()
527 work_unit_size, shard); in Compute()
Dconv_grad_ops_3d.cc397 const int64 work_unit_size = size_A + size_B + size_C; in Compute() local
411 : (target_working_set_size + work_unit_size - 1) / work_unit_size; in Compute()
564 work_unit_size, shard); in Compute()
870 const int64 work_unit_size = size_A + size_B + size_C; in Compute() local
873 (target_working_set_size + work_unit_size - 1) / work_unit_size; in Compute()
Dconv_grad_filter_ops.cc320 const size_t work_unit_size = size_A + size_B + size_C; in Compute() local
323 (target_working_set_size + work_unit_size - 1) / work_unit_size; in Compute()
Dpooling_ops_common.h592 const int64 work_unit_size =
599 const int64 work_unit_cost = std::max(int64{10000}, work_unit_size / 100LL);