Searched defs:cost_per_unit (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | work_sharder.cc | 33 int64 cost_per_unit, std::function<void(int64, int64)> work) { in Shard() argument
55 void Sharder::Do(int64 total, int64 cost_per_unit, const Work& work, in Do()
|
| D | work_sharder_test.cc | 30 void RunSharding(int64 num_workers, int64 total, int64 cost_per_unit, in RunSharding()
82 const int64 cost_per_unit = 10; in TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_cpu.h | 36 int64 cost_per_unit, ElementCopier copier, in ConcatCPUImpl()
138 int64 cost_per_unit, ElementCopier copier, in ConcatSYCLImpl()
|
| D | reverse_op.cc | 79 const int64 cost_per_unit = input.NumElements() / N; in ReverseRows() local
|
| /external/tensorflow/tensorflow/contrib/nearest_neighbor/kernels/ |
| D | hyperplane_lsh_probes.cc | 119 int64 cost_per_unit = 21 * num_hyperplanes_per_table * num_tables; in Compute() local
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | threadpool.cc | 179 void ThreadPool::ParallelFor(int64 total, int64 cost_per_unit, in ParallelFor()
185 int64 total, int64 cost_per_unit, in ParallelForWithWorkerId() argument
|
| D | threadpool_test.cc | 239 static void BM_ParallelFor(int iters, int total, int cost_per_unit) { in BM_ParallelFor()
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | masked_matmul_ops.cc | 173 const int64 cost_per_unit = 20 * a_dim_1; in Compute() local
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | pmf_to_cdf_op.cc | 68 const int64 cost_per_unit = static_cast<int64>(50.0 * n * std::log2(n)); in Compute() local
|