| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/quantiles/ |
| D | weighted_quantiles_stream_test.cc | 58 void GenerateFixedUniformSummary(int32 worker_id, int64 max_elements, in GenerateFixedUniformSummary() argument
68 void GenerateFixedNonUniformSummary(int32 worker_id, int64 max_elements, in GenerateFixedNonUniformSummary() argument
78 void GenerateRandUniformFixedWeightsSummary(int32 worker_id, int64 max_elements, in GenerateRandUniformFixedWeightsSummary() argument
82 random::PhiloxRandom philox(13 + worker_id); in GenerateRandUniformFixedWeightsSummary()
92 void GenerateRandUniformRandWeightsSummary(int32 worker_id, int64 max_elements, in GenerateRandUniformRandWeightsSummary() argument
96 random::PhiloxRandom philox(13 + worker_id); in GenerateRandUniformRandWeightsSummary()
134 void GenerateOneValue(int32 worker_id, int64 max_elements, double *total_weight, in GenerateOneValue() argument
141 void GenerateOneZeroWeightedValue(int32 worker_id, int64 max_elements, in GenerateOneZeroWeightedValue() argument
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/ |
| D | weighted_quantiles_stream_test.cc | 57 void GenerateFixedUniformSummary(int32 worker_id, int64 max_elements, in GenerateFixedUniformSummary() argument
67 void GenerateFixedNonUniformSummary(int32 worker_id, int64 max_elements, in GenerateFixedNonUniformSummary() argument
77 void GenerateRandUniformFixedWeightsSummary(int32 worker_id, int64 max_elements, in GenerateRandUniformFixedWeightsSummary() argument
81 random::PhiloxRandom philox(13 + worker_id); in GenerateRandUniformFixedWeightsSummary()
91 void GenerateRandUniformRandWeightsSummary(int32 worker_id, int64 max_elements, in GenerateRandUniformRandWeightsSummary() argument
95 random::PhiloxRandom philox(13 + worker_id); in GenerateRandUniformRandWeightsSummary()
133 void GenerateOneValue(int32 worker_id, int64 max_elements, double *total_weight, in GenerateOneValue() argument
140 void GenerateOneZeroWeightedValue(int32 worker_id, int64 max_elements, in GenerateOneZeroWeightedValue() argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | sync_replicas_optimizer_test.py | 39 for worker_id in range(num_workers):
41 is_chief = (worker_id == 0)
51 with ops.device("/job:worker/task:" + str(worker_id)):
52 grads_0 = constant_op.constant(0.1 + worker_id * 0.2)
53 grads_1 = constant_op.constant(0.9 + worker_id * 0.2)
57 [0.1 + worker_id * 0.2], shape=[1, 1]),
76 master=workers[worker_id].target,
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | model_average_optimizer_test.py | 65 for worker_id in range(num_workers):
67 is_chief = (worker_id == 0)
69 worker_device = "/job:worker/task:%d/cpu:0" % (worker_id)
83 with ops.device("/job:worker/task:" + str(worker_id)):
84 if worker_id == 0:
104 workers[worker_id].target, hooks=[ma_hook])
|
| D | agn_optimizer_test.py | 68 for worker_id in range(num_workers):
70 is_chief = (worker_id == 0)
72 worker_device = "/job:worker/task:%d/cpu:0" % (worker_id)
97 with ops.device("/job:worker/task:" + str(worker_id)):
121 hook = opt.make_session_run_hook(is_chief, worker_id)
124 workers[worker_id].target, hooks=[hook])
|
| D | elastic_average_optimizer_test.py | 72 for worker_id in range(num_workers):
74 is_chief = (worker_id == 0)
76 worker_device = "/job:worker/task:%d/cpu:0" % (worker_id)
105 with ops.device("/job:worker/task:" + str(worker_id)):
131 easgd_hook = opt.make_session_run_hook(is_chief, worker_id)
135 workers[worker_id].target, hooks=[easgd_hook])
|
| D | drop_stale_gradient_optimizer_test.py | 60 for worker_id in range(num_workers):
81 if worker_id == 0:
91 if worker_id == 0:
103 sess = session.Session(workers[worker_id].target)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bincount_op.cc | 66 [&](int64 start_ind, int64 limit_ind, int64 worker_id) { in Compute() argument
71 partial_bins(worker_id, value) += weights(i); in Compute()
74 partial_bins(worker_id, value) += T(1); in Compute()
|
| /external/libaom/libaom/av1/decoder/ |
| D | dthread.h | 42 int worker_id; member
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/kernel_tests/ |
| D | sdca_ops_test.py | 491 def minimize(worker_id): argument
494 str(i + worker_id*len(example_weights)) for i in range(
500 for worker_id in range(num_loss_partitions):
501 threads.append(threading.Thread(target=minimize, args=(worker_id,)))
|
| /external/libaom/libaom/av1/ |
| D | av1_dx_iface.c | 480 frame_worker_data->worker_id = i; in init_decoder()
|