Home
last modified time | relevance | path

Searched refs:iter_num (Results 1 – 25 of 26) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/
Dexchange_keys_kernel.cc41 …sKernel::ReachThresholdForExchangeKeys(const std::shared_ptr<FBBuilder> &fbb, const int iter_num) { in ReachThresholdForExchangeKeys() argument
47 IntToSize(iter_num)); in ReachThresholdForExchangeKeys()
56 const int iter_num) { in CountForExchangeKeys() argument
63 IntToSize(iter_num)); in CountForExchangeKeys()
74 size_t iter_num = LocalMetaStore::GetInstance().curr_iter_num(); in Launch() local
76 …MS_LOG(INFO) << "ITERATION NUMBER IS : " << iter_num << ", Total ExchangeKeysKernel allowed Durati… in Launch()
94 if (ReachThresholdForExchangeKeys(fbb, SizeToInt(iter_num))) { in Launch()
100 if (iter_num != iter_client) { in Launch()
101 … MS_LOG(ERROR) << "ExchangeKeys iteration number is invalid: server now iteration is " << iter_num in Launch()
104 std::to_string(CURRENT_TIME_MILLI.count()), iter_num); in Launch()
[all …]
Dshare_secrets_kernel.cc41 const size_t iter_num) { in CountForShareSecrets() argument
45 … std::to_string(CURRENT_TIME_MILLI.count()), SizeToInt(iter_num)); in CountForShareSecrets()
55 size_t iter_num = LocalMetaStore::GetInstance().curr_iter_num(); in Launch() local
57 …MS_LOG(INFO) << "ITERATION NUMBER IS : " << iter_num << ", Total ShareSecretsKernel allowed Durati… in Launch()
79 … std::to_string(CURRENT_TIME_MILLI.count()), SizeToInt(iter_num)); in Launch()
85 if (iter_num != iter_client) { in Launch()
86 MS_LOG(ERROR) << "ShareSecretsKernel iteration invalid. server now iteration is " << iter_num in Launch()
89 … std::to_string(CURRENT_TIME_MILLI.count()), SizeToInt(iter_num)); in Launch()
93 response = cipher_share_->ShareSecrets(SizeToInt(iter_num), share_secrets_req, fbb, in Launch()
100 if (!CountForShareSecrets(fbb, share_secrets_req, iter_num)) { in Launch()
Dclient_list_kernel.cc42 bool ClientListKernel::DealClient(const size_t iter_num, const schema::GetClientList *get_clients_r… in DealClient() argument
51 … empty_client_list, std::to_string(CURRENT_TIME_MILLI.count()), SizeToInt(iter_num)); in DealClient()
64 std::to_string(CURRENT_TIME_MILLI.count()), SizeToInt(iter_num)); in DealClient()
72 std::to_string(CURRENT_TIME_MILLI.count()), SizeToInt(iter_num)); in DealClient()
82 std::to_string(CURRENT_TIME_MILLI.count()), SizeToInt(iter_num)); in DealClient()
88 std::to_string(CURRENT_TIME_MILLI.count()), SizeToInt(iter_num)); in DealClient()
99 std::to_string(CURRENT_TIME_MILLI.count()), SizeToInt(iter_num)); in DealClient()
105 size_t iter_num = LocalMetaStore::GetInstance().curr_iter_num(); in Launch() local
107 …MS_LOG(INFO) << "Iteration number is " << iter_num << ", ClientListKernel total duration is " << t… in Launch()
126 if (iter_num != iter_client) { in Launch()
[all …]
Dget_keys_kernel.cc41 const int iter_num) { in CountForGetKeys() argument
46 fbb, schema::ResponseCode_OutOfTime, IntToSize(iter_num), in CountForGetKeys()
58 size_t iter_num = LocalMetaStore::GetInstance().curr_iter_num(); in Launch() local
60 …MS_LOG(INFO) << "ITERATION NUMBER IS : " << iter_num << ", Total GetKeysKernel allowed Duration Is… in Launch()
84 if (iter_num != iter_client) { in Launch()
85 MS_LOG(ERROR) << "GetKeysKernel iteration invalid. server now iteration is " << iter_num in Launch()
87 cipher_key_->BuildGetKeysRsp(fbb, schema::ResponseCode_OutOfTime, iter_num, in Launch()
92 …response = cipher_key_->GetKeys(iter_num, std::to_string(CURRENT_TIME_MILLI.count()), get_exchange… in Launch()
98 if (!CountForGetKeys(fbb, get_exchange_keys_req, SizeToInt(iter_num))) { in Launch()
Dget_secrets_kernel.cc45 … const schema::GetShareSecrets *get_secrets_req, const int iter_num) { in CountForGetSecrets() argument
49 cipher_share_->BuildGetSecretsRsp(fbb, schema::ResponseCode_OutOfTime, IntToSize(iter_num), in CountForGetSecrets()
60 size_t iter_num = LocalMetaStore::GetInstance().curr_iter_num(); in Launch() local
64 …MS_LOG(INFO) << "ITERATION NUMBER IS : " << iter_num << ", Total GetSecretsKernel allowed Duration… in Launch()
84 if (iter_num != iter_client) { in Launch()
85 MS_LOG(ERROR) << "GetSecretsKernel iteration invalid. server now iteration is " << iter_num in Launch()
87 …cipher_share_->BuildGetSecretsRsp(fbb, schema::ResponseCode_OutOfTime, iter_num, next_timestamp, n… in Launch()
102 if (!CountForGetSecrets(fbb, get_secrets_req, SizeToInt(iter_num))) { in Launch()
Dreconstruct_secrets_kernel.cc50 size_t iter_num = LocalMetaStore::GetInstance().curr_iter_num(); in Launch() local
52 …MS_LOG(INFO) << "Iteration number is " << iter_num << ", ReconstructSecretsKernel total duration i… in Launch()
90 … SizeToInt(iter_num), std::to_string(CURRENT_TIME_MILLI.count())); in Launch()
94 … SizeToInt(iter_num), std::to_string(CURRENT_TIME_MILLI.count())); in Launch()
100 …response = cipher_reconstruct_.ReconstructSecrets(SizeToInt(iter_num), std::to_string(CURRENT_TIME… in Launch()
Dexchange_keys_kernel.h46 bool ReachThresholdForExchangeKeys(const std::shared_ptr<FBBuilder> &fbb, const int iter_num);
48 const int iter_num);
Dget_secrets_kernel.h46 const int iter_num);
Dshare_secrets_kernel.h47 const size_t iter_num);
Dget_keys_kernel.h47 const int iter_num);
Dclient_list_kernel.h46 bool DealClient(const size_t iter_num, const schema::GetClientList *get_clients_req,
/third_party/mindspore/tests/st/data_transfer/
Dtest_tdt_data_transfer.py55 iter_num = 0
61 iter_num += 1
62 logger.info("Iter Num : %s", iter_num)
64 return iter_num
143 iter_num = op_network_with_step_num(ds, step_num=beyond_step_num)
144 logger.info("out_iter_num：%s", iter_num)
164 iter_num = op_network_with_step_num(ds, step_num=beyond_step_num)
165 logger.info("out_iter_num：%s", iter_num)
166 assert iter_num == 10
/third_party/boost/boost/graph/distributed/
Dboman_et_al_graph_coloring.hpp156 iterations_type iter_num = 1, outer_iter_num = 1; in boman_et_al_graph_coloring() local
190 marked[k] = iter_num; in boman_et_al_graph_coloring()
195 put(color, *first, choose_color(marked, iter_num)); in boman_et_al_graph_coloring()
211 ++iter_num; in boman_et_al_graph_coloring()
303 if (marked[k] != iter_num) { in boman_et_al_graph_coloring()
304 marked[k] = iter_num; in boman_et_al_graph_coloring()
/third_party/mindspore/mindspore/ccsrc/ps/core/protos/
Dfl.proto130 int32 iter_num = 3; field
192 uint64 iter_num = 3; field
/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/
Dtest_fl_fasterrcnn.py204 for iter_num in range(fl_iteration_num):
212 ckpt_name = user + "-fast-rcnn-" + str(iter_num) + "epoch.ckpt"
/third_party/mindspore/tests/st/fl/cross_silo_femnist/
Dtest_cross_silo_femnist.py327 for iter_num in range(fl_iteration_num):
340 ckpt_name = user_id + "-fl-ms-bs32-" + str(iter_num) + "epoch.ckpt"
348 …print('local epoch: {}, loss: {}, trian acc: {}, test acc: {}'.format(iter_num, loss, train_acc, t…
/third_party/mindspore/mindspore/ccsrc/fl/server/
Diteration.cc423 const auto &iter_num = notify_leader_to_next_iter_req.iter_num(); in HandleNotifyLeaderMoveToNextIterRequest() local
426 << ". Iteration number: " << iter_num << ". Reason: " << reason; in HandleNotifyLeaderMoveToNextIterRequest()
428 if (IsMoveToNextIterRequestReentrant(iter_num)) { in HandleNotifyLeaderMoveToNextIterRequest()
/third_party/mindspore/mindspore/ccsrc/utils/
Dconfig_manager.h89 int64_t iter_num() const { in iter_num() function
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline.cc161 int64_t loop_size = ConfigManager::GetInstance().iter_num(); in SetLoopCount()
1147 bool InitExecDataset(const std::string &queue_name, int64_t iter_num, int64_t batch_size, in InitExecDataset() argument
1158 if (iter_num == -1) { in InitExecDataset()
1159 iter_num = INT32_MAX; in InitExecDataset()
1162 …return InitExecDatasetVm(queue_name, iter_num, batch_size, types, shapes, input_indexes, need_run); in InitExecDataset()
1165 return InitExecDatasetGe(queue_name, iter_num, batch_size, types, shapes, input_indexes, phase); in InitExecDataset()
Dpipeline.h166 bool InitExecDataset(const std::string &queue_name, int64_t iter_num, int64_t batch_size,
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_backend_optimization.cc305 ConfigManager::GetInstance().iter_num() > 1) { in AscendBackendIRFusionOptimization()
435 ConfigManager::GetInstance().iter_num() > 1) { in AscendBackendOptimization()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc221 auto iter_num = in FilterInVaildShape() local
223 for (size_t i = 0; i < iter_num; ++i) { in FilterInVaildShape()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc70 …text_ptr->get_param<bool>(MS_CTX_ENABLE_LOOP_SINK) && ConfigManager::GetInstance().iter_num() > 1); in NeedInsertSwitch()
428 bool eos_mode = ConfigManager::GetInstance().iter_num() == INT32_MAX && exist_getnext; in InsertSwitchLoop()
430 << " iter num:" << ConfigManager::GetInstance().iter_num(); in InsertSwitchLoop()
805 *val = SizeToInt(LongToSize(ConfigManager::GetInstance().iter_num())); in LoadSwitchInputs()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.cc156 value = ConfigManager::GetInstance().iter_num(); in InitLoopVar()
939 if (ConfigManager::GetInstance().iter_num() > 1) { in BuildGraph()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc453 …r::GetInstance().dataset_mode() == DS_SINK_MODE && (ConfigManager::GetInstance().iter_num() > 1)) { in GenTask()

12