| /external/gemmlowp/test/ |
| D | test_blocking_counter.cc | 26 Thread(BlockingCounter* blocking_counter, int number_of_times_to_decrement) in Thread()
59 BlockingCounter* const blocking_counter_;
66 void test_blocking_counter(BlockingCounter* blocking_counter, int num_threads, in test_blocking_counter()
87 BlockingCounter* blocking_counter = new BlockingCounter; in test_blocking_counter()
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | blocking_counter_test.cc | 25 BlockingCounter bc(0); in TEST()
30 BlockingCounter bc(2); in TEST()
41 BlockingCounter bc(N); in TEST()
60 BlockingCounter bc(num_shards); in BM_BlockingCounter()
|
| D | blocking_counter.h | 26 class BlockingCounter {
28 BlockingCounter(int initial_count) in BlockingCounter() function
34 ~BlockingCounter() {} in ~BlockingCounter()
|
| D | threadpool.cc | 153 BlockingCounter counter(num_shards_used); in TransformRangeConcurrently()
|
| /external/gemmlowp/internal/ |
| D | multi_thread_gemm.h | 150 class BlockingCounter {
152 BlockingCounter() : count_(0), initial_count_(0) { in BlockingCounter() function
157 ~BlockingCounter() { in ~BlockingCounter()
237 explicit Worker(BlockingCounter* counter_to_decrement_when_ready) in Worker()
353 BlockingCounter* const counter_to_decrement_when_ready_;
419 BlockingCounter counter_to_decrement_when_ready_;
|
| /external/tensorflow/tensorflow/cc/training/ |
| D | coordinator_test.cc | 76 void StartSettingStatus(const Status& status, BlockingCounter* counter, in StartSettingStatus()
107 void SetStatusThread(const Status& status, BlockingCounter* counter, in SetStatusThread()
179 BlockingCounter counter(3); in TEST()
|
| D | queue_runner.h | 124 std::unique_ptr<BlockingCounter> counter_;
|
| D | queue_runner.cc | 97 counter_.reset(new BlockingCounter(runs_)); in Start()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 815 static inline std::unique_ptr<BlockingCounter> CreateSparseSlices(
825 static inline std::unique_ptr<BlockingCounter> CreateDenseSlices(
833 static inline BlockingCounter* ShuffleMatrix(
1137 inline std::unique_ptr<BlockingCounter>
1150 BlockingCounter* counter = in CreateSparseSlices()
1151 new BlockingCounter(num_slices_dim0 * num_slices_dim1); in CreateSparseSlices()
1186 return std::unique_ptr<BlockingCounter>(counter); in CreateSparseSlices()
1234 inline BlockingCounter* SparseMatMul<TL, TR>::ShuffleMatrix( in ShuffleMatrix()
1242 BlockingCounter* counter = new BlockingCounter(num_threads); in ShuffleMatrix()
1303 inline std::unique_ptr<BlockingCounter> SparseMatMul<TL, TR>::CreateDenseSlices( in CreateDenseSlices()
[all …]
|
| D | xsmm_conv2d.cc | 238 BlockingCounter count(num_threads); in CallLibxsmmConvGeneric()
256 BlockingCounter count(num_threads); in CallLibxsmmConvGeneric()
368 BlockingCounter counter(num_threads); in CallLibxsmmConvGeneric()
|
| D | clustering_ops.cc | 412 BlockingCounter counter(num_threads - 1); in Compute()
|
| D | quantization_utils.h | 937 gemmlowp::BlockingCounter counter_to_decrement_when_ready_;
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | parallel_for.cc | 35 BlockingCounter counter(num_shards_used - 1); in ParallelFor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fork_join.cc | 85 tensorflow::BlockingCounter bc(num_partitions - 1); in __xla_cpu_runtime_ParallelForkJoin()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | work_sharder.cc | 82 BlockingCounter counter(num_shards_used - 1); in Do()
|
| D | example_proto_fast_parsing.cc | 436 BlockingCounter counter(n - 1); in ParallelFor()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | measuring_cost_estimator.cc | 63 BlockingCounter barrier(measurement_steps_); in PredictCosts()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | unbounded_thread_pool_test.cc | 118 BlockingCounter bc(round_size); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 172 BlockingCounter counter(static_cast<int>(remote_contexts_.size())); in CloseRemoteContexts()
310 BlockingCounter blocking_counter(static_cast<int>(remote_contexts_.size())); in MaybeRegisterFunctionRemotely()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_all_reduce_thunk.cc | 145 absl::optional<tensorflow::BlockingCounter> done_;
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | wals_solver_ops.cc | 213 BlockingCounter counter(shards.size()); in Compute()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_session.cc | 165 BlockingCounter all_done(waiting_for); in RetrieveLogs()
466 BlockingCounter done(num); in DoRegisterPartitions()
539 BlockingCounter pending_;
1250 BlockingCounter done(worker_names.size()); in CreateWorkerSessions()
1332 BlockingCounter done(worker_names.size()); in DeleteWorkerSessions()
|
| /external/tensorflow/tensorflow/lite/tools/accuracy/ilsvrc/ |
| D | imagenet_model_evaluator.cc | 313 BlockingCounter counter(num_threads_); in EvaluateModel()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | matching_files_dataset_op.cc | 319 BlockingCounter counter(children.size()); in UpdateIterator()
|
| /external/tensorflow/tensorflow/core/platform/cloud/ |
| D | ram_file_block_cache_test.cc | 480 BlockingCounter counter(callers); in TEST()
|