HomeSort by relevance Sort by last modified time
    Searched refs:num_devices_per_task (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/core/framework/
collective.cc 68 num_devices_per_task = other.num_devices_per_task;
98 strings::StrAppend(&v, "} num_devices_per_task={");
99 for (const auto dpt : num_devices_per_task) {
collective.h 92 std::unordered_map<string, int32> num_devices_per_task; member in struct:tensorflow::CollInstanceParams
  /external/tensorflow/tensorflow/core/common_runtime/
collective_param_resolver_local.cc 322 cp->instance.num_devices_per_task.clear();
329 cp->instance.num_devices_per_task[*last_task_name] = count;
334 cp->instance.num_devices_per_task[*last_task_name] = count;
338 for (const auto& task_dev : cp->instance.num_devices_per_task) {
base_collective_executor.cc 339 col_params.instance.num_devices_per_task.at(task_name);
  /external/tensorflow/tensorflow/core/kernels/
collective_nccl_reducer.cc 136 const int num_local_devices = col_params_->instance.num_devices_per_task.at(
collective_nccl_reducer_test.cc 127 col_params_.instance.num_devices_per_task[task_name] = num_ranks;

Completed in 910 milliseconds