Searched defs:cost_per_unit (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/platform/ |
| D | threadpool.h | 77 absl::optional<int64> cost_per_unit, in SchedulingParams()
84 absl::optional<int64> cost_per_unit() const { return cost_per_unit_; } in cost_per_unit() function
|
| D | threadpool.cc | 203 void ThreadPool::ParallelFor(int64 total, int64 cost_per_unit, in ParallelFor()
213 int64 total, int64 cost_per_unit, in ParallelForWithWorkerId() argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | work_sharder.cc | 33 int64 cost_per_unit, std::function<void(int64, int64)> work) { in Shard() argument
57 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.cc | 104 int64 cost_per_unit = EstimateBytesPerElement<T>(inputs); in ConcatCPU() local
|
| D | concat_lib_cpu.h | 36 int64 cost_per_unit, ElementCopier copier, in ConcatCPUImpl()
|
| D | reverse_op.cc | 76 const int64 cost_per_unit = input.NumElements() / N; in ReverseRows() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | banded_triangular_solve_op.cc | 173 const int64 cost_per_unit = in Launch() local
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | threadpool_test.cc | 413 static void BM_ParallelFor(int iters, int total, int cost_per_unit) { in BM_ParallelFor()
|