Searched refs:local_rank (Results 1 – 8 of 8) sorted by relevance
130 uint32_t local_rank = 0; in GetDevicesByTensorMap() local135 ++local_rank; in GetDevicesByTensorMap()137 if (local_rank == dev_list_.size()) { in GetDevicesByTensorMap()138 MS_LOG(ERROR) << "Rank id: " << local_rank << "is not in the device list."; in GetDevicesByTensorMap()142 Shape current_rank_coordinate = ConvertRankToCoordinate((int32_t)local_rank, dev_shape_); in GetDevicesByTensorMap()
317 local_rank = int(_get_global_rank() % (_get_device_num() / _get_pipeline_stages()))319 local_rank = _get_global_rank()343 ret[param_name] = ([local_rank], True)349 ret[param_name] = ([rank_list[local_rank]], True)
134 int local_rank() const { return this->rank(); } in local_rank() function in boost::mpi::intercommunicator
85 uint32_t local_rank() { return local_rank_; } in local_rank() function
25 …csKernel::InitKernel(size_t) { local_rank_ = DistributedCountService::GetInstance().local_rank(); } in InitKernel()
131 std::string worker_id = std::to_string(DistributedCountService::GetInstance().local_rank()); in OnLastCountEvent()
30 local_rank_ = DistributedCountService::GetInstance().local_rank(); in InitKernel()
154 …name_, std::to_string(DistributedCountService::GetInstance().local_rank()) + "_" + std::to_string(… in Launch()