OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:num_local_devices
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/core/nccl/
nccl_manager.cc
137
num_local_devices
(num_local_devices_in),
148
const int
num_local_devices
; // devices local to this node
member in struct:tensorflow::NcclManager::Collective
166
// eligible for running with <available_participants> ==
num_local_devices
.
230
for (i = 0; i < collective->
num_local_devices
; ++i) {
236
if (i == collective->
num_local_devices
) {
269
std::vector<CommunicatorMember> members(collective->
num_local_devices
);
270
std::vector<int> devices(collective->
num_local_devices
);
271
for (int i = 0; i < collective->
num_local_devices
; ++i) {
301
std::vector<ncclComm_t> nccl_comms(collective->
num_local_devices
);
314
for (int i = 0; i < collective->
num_local_devices
; ++i)
[
all
...]
nccl_manager.h
117
Context(const string& collective_key, int
num_local_devices
,
120
num_local_devices
(
num_local_devices
),
128
int
num_local_devices
;
member in struct:tensorflow::NcclManager::Context
/external/tensorflow/tensorflow/core/kernels/
collective_nccl_reducer.cc
136
const int
num_local_devices
= col_params_->instance.num_devices_per_task.at(
local
151
<< " num local devices " <<
num_local_devices
157
{nccl_collective_key,
num_local_devices
, num_global_devices,
/external/tensorflow/tensorflow/core/distributed_runtime/
master.cc
451
int
num_local_devices
= 0;
454
if (
num_local_devices
== 0) {
458
num_local_devices
++;
Completed in 393 milliseconds