Home
last modified time | relevance | path

Searched refs:FBBuilder (Results 1 – 25 of 37) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/
Dstart_fl_job_kernel.h47 ResultCode ReachThresholdForStartFLJob(const std::shared_ptr<FBBuilder> &fbb);
54 …ResultCode ReadyForStartFLJob(const std::shared_ptr<FBBuilder> &fbb, const DeviceMeta &device_meta…
57 …ResultCode CountForStartFLJob(const std::shared_ptr<FBBuilder> &fbb, const schema::RequestFLJob *s…
59 void StartFLJob(const std::shared_ptr<FBBuilder> &fbb, const DeviceMeta &device_meta);
62 void BuildStartFLJobRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::ResponseCode retcode,
Dupdate_model_kernel.h53 ResultCode ReachThresholdForUpdateModel(const std::shared_ptr<FBBuilder> &fbb);
54 …teModel(const schema::RequestUpdateModel *update_model_req, const std::shared_ptr<FBBuilder> &fbb);
56 ResultCode CountForUpdateModel(const std::shared_ptr<FBBuilder> &fbb,
58 …void BuildUpdateModelRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::ResponseCode retcode,
Dpush_metrics_kernel.cc31 std::shared_ptr<FBBuilder> fbb = std::make_shared<FBBuilder>(); in Launch()
76 ResultCode PushMetricsKernel::PushMetrics(const std::shared_ptr<FBBuilder> &fbb, in PushMetrics()
98 void PushMetricsKernel::BuildPushMetricsRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::Re… in BuildPushMetricsRsp()
Dexchange_keys_kernel.cc41 bool ExchangeKeysKernel::ReachThresholdForExchangeKeys(const std::shared_ptr<FBBuilder> &fbb, const… in ReachThresholdForExchangeKeys()
54 bool ExchangeKeysKernel::CountForExchangeKeys(const std::shared_ptr<FBBuilder> &fbb, in CountForExchangeKeys()
87 std::shared_ptr<server::FBBuilder> fbb = std::make_shared<server::FBBuilder>(); in Launch()
Dstart_fl_job_kernel.cc62 std::shared_ptr<FBBuilder> fbb = std::make_shared<FBBuilder>(); in Launch()
140 ResultCode StartFLJobKernel::ReachThresholdForStartFLJob(const std::shared_ptr<FBBuilder> &fbb) { in ReachThresholdForStartFLJob()
166 ResultCode StartFLJobKernel::ReadyForStartFLJob(const std::shared_ptr<FBBuilder> &fbb, const Device… in ReadyForStartFLJob()
182 ResultCode StartFLJobKernel::CountForStartFLJob(const std::shared_ptr<FBBuilder> &fbb, in CountForStartFLJob()
197 void StartFLJobKernel::StartFLJob(const std::shared_ptr<FBBuilder> &fbb, const DeviceMeta &) { in StartFLJob()
209 void StartFLJobKernel::BuildStartFLJobRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::Resp… in BuildStartFLJobRsp()
Dpull_weight_kernel.cc41 std::shared_ptr<FBBuilder> fbb = std::make_shared<FBBuilder>(); in Launch()
66 void PullWeightKernel::PullWeight(const std::shared_ptr<FBBuilder> &fbb, in PullWeight()
118 void PullWeightKernel::BuildPullWeightRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::Resp… in BuildPullWeightRsp()
Dget_model_kernel.cc52 std::shared_ptr<FBBuilder> fbb = std::make_shared<FBBuilder>(); in Launch()
94 …el::GetModel(const schema::RequestGetModel *get_model_req, const std::shared_ptr<FBBuilder> &fbb) { in GetModel()
130 void GetModelKernel::BuildGetModelRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::Response… in BuildGetModelRsp()
Dexchange_keys_kernel.h46 bool ReachThresholdForExchangeKeys(const std::shared_ptr<FBBuilder> &fbb, const int iter_num);
47 …bool CountForExchangeKeys(const std::shared_ptr<FBBuilder> &fbb, const schema::RequestExchangeKeys…
Dpush_metrics_kernel.h45 …ResultCode PushMetrics(const std::shared_ptr<FBBuilder> &fbb, const schema::RequestPushMetrics *pu…
46 …void BuildPushMetricsRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::ResponseCode retcode…
Dpush_weight_kernel.cc37 std::shared_ptr<FBBuilder> fbb = std::make_shared<FBBuilder>(); in Launch()
81 ResultCode PushWeightKernel::PushWeight(const std::shared_ptr<FBBuilder> &fbb, in PushWeight()
136 void PushWeightKernel::BuildPushWeightRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::Resp… in BuildPushWeightRsp()
Dclient_list_kernel.h40 void BuildClientListRsp(const std::shared_ptr<server::FBBuilder> &client_list_resp_builder,
47 const std::shared_ptr<server::FBBuilder> &fbb);
Dget_secrets_kernel.cc44 bool GetSecretsKernel::CountForGetSecrets(const std::shared_ptr<FBBuilder> &fbb, in CountForGetSecrets()
74 std::shared_ptr<server::FBBuilder> fbb = std::make_shared<server::FBBuilder>(); in Launch()
Dget_keys_kernel.cc40 bool GetKeysKernel::CountForGetKeys(const std::shared_ptr<FBBuilder> &fbb, const schema::GetExchang… in CountForGetKeys()
70 std::shared_ptr<server::FBBuilder> fbb = std::make_shared<server::FBBuilder>(); in Launch()
Dpull_weight_kernel.h45 …void PullWeight(const std::shared_ptr<FBBuilder> &fbb, const schema::RequestPullWeight *pull_weigh…
46 void BuildPullWeightRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::ResponseCode retcode,
Dget_model_kernel.h45 …void GetModel(const schema::RequestGetModel *get_model_req, const std::shared_ptr<FBBuilder> &fbb);
46 void BuildGetModelRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::ResponseCode retcode,
Dupdate_model_kernel.cc56 std::shared_ptr<FBBuilder> fbb = std::make_shared<FBBuilder>(); in Launch()
130 ResultCode UpdateModelKernel::ReachThresholdForUpdateModel(const std::shared_ptr<FBBuilder> &fbb) { in ReachThresholdForUpdateModel()
143 const std::shared_ptr<FBBuilder> &fbb) { in UpdateModel()
246 ResultCode UpdateModelKernel::CountForUpdateModel(const std::shared_ptr<FBBuilder> &fbb, in CountForUpdateModel()
262 void UpdateModelKernel::BuildUpdateModelRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::Re… in BuildUpdateModelRsp()
Dpush_weight_kernel.h45 …ResultCode PushWeight(const std::shared_ptr<FBBuilder> &fbb, const schema::RequestPushWeight *push…
47 void BuildPushWeightRsp(const std::shared_ptr<FBBuilder> &fbb, const schema::ResponseCode retcode,
Dshare_secrets_kernel.cc39 bool ShareSecretsKernel::CountForShareSecrets(const std::shared_ptr<FBBuilder> &fbb, in CountForShareSecrets()
67 std::shared_ptr<server::FBBuilder> fbb = std::make_shared<server::FBBuilder>(); in Launch()
Dclient_list_kernel.cc43 const std::shared_ptr<server::FBBuilder> &fbb) { in DealClient()
116 std::shared_ptr<server::FBBuilder> fbb = std::make_shared<server::FBBuilder>(); in Launch()
156 void ClientListKernel::BuildClientListRsp(const std::shared_ptr<server::FBBuilder> &client_list_res… in BuildClientListRsp()
/third_party/mindspore/mindspore/ccsrc/fl/armour/cipher/
Dcipher_shares.h47 const std::shared_ptr<fl::server::FBBuilder> &share_secrets_resp_builder,
51 const std::shared_ptr<fl::server::FBBuilder> &get_secrets_resp_builder,
55 …void BuildShareSecretsRsp(const std::shared_ptr<fl::server::FBBuilder> &share_secrets_resp_builder,
59 …void BuildGetSecretsRsp(const std::shared_ptr<fl::server::FBBuilder> &fbb, const schema::ResponseC…
Dcipher_keys.h48 …schema::GetExchangeKeys *get_exchange_keys_req, const std::shared_ptr<fl::server::FBBuilder> &fbb);
53 const std::shared_ptr<fl::server::FBBuilder> &fbb);
56 …void BuildGetKeysRsp(const std::shared_ptr<fl::server::FBBuilder> &fbb, const schema::ResponseCode…
59 …void BuildExchangeKeysRsp(const std::shared_ptr<fl::server::FBBuilder> &fbb, const schema::Respons…
Dcipher_reconstruct.h50 const std::shared_ptr<fl::server::FBBuilder> &fbb,
54 …void BuildReconstructSecretsRsp(const std::shared_ptr<fl::server::FBBuilder> &fbb, const schema::R…
89 const std::shared_ptr<fl::server::FBBuilder> &fbb, const int cur_iterator,
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/
Dpush_metrics_kernel.h100 fbb_ = std::make_shared<fl::FBBuilder>(); in Init()
113 bool BuildPushMetricsReq(const std::shared_ptr<fl::FBBuilder> &fbb, T loss, T accuracy) { in BuildPushMetricsReq()
123 std::shared_ptr<fl::FBBuilder> fbb_;
Dstart_fl_job_kernel.h88 fbb_ = std::make_shared<fl::FBBuilder>(); in Init()
101 bool BuildStartFLJobReq(const std::shared_ptr<fl::FBBuilder> &fbb) { in BuildStartFLJobReq()
120 std::shared_ptr<fl::FBBuilder> fbb_;
Dfused_pull_weight_kernel.h49 std::shared_ptr<fl::FBBuilder> fbb = std::make_shared<fl::FBBuilder>(); in Launch()
96 fbb = std::make_shared<fl::FBBuilder>(); in Launch()
159 bool BuildPullWeightReq(std::shared_ptr<fl::FBBuilder> fbb) { in BuildPullWeightReq()

12