Home
last modified time | relevance | path

Searched refs:num_devices_per_task (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dcollective.cc68 num_devices_per_task = other.num_devices_per_task; in operator =()
99 for (const auto dpt : num_devices_per_task) { in ToString()
Dcollective.h92 std::unordered_map<string, int32> num_devices_per_task; member
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc322 cp->instance.num_devices_per_task.clear(); in SetDevPerTask()
329 cp->instance.num_devices_per_task[*last_task_name] = count; in SetDevPerTask()
334 cp->instance.num_devices_per_task[*last_task_name] = count; in SetDevPerTask()
338 for (const auto& task_dev : cp->instance.num_devices_per_task) { in SetDevPerTask()
Dbase_collective_executor.cc339 col_params.instance.num_devices_per_task.at(task_name); in Launched()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc136 const int num_local_devices = col_params_->instance.num_devices_per_task.at( in Run()
Dcollective_nccl_reducer_test.cc127 col_params_.instance.num_devices_per_task[task_name] = num_ranks; in Init()