Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_inference_session.cc42 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in LoadInputData() local
43 MS_EXCEPTION_IF_NULL(pk_node); in LoadInputData()
44 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in LoadInputData()
46 if (!AnfAlgo::IsParameterWeight(pk_node)) { in LoadInputData()
48 if (!device_address->SyncHostToDevice(trans::GetRuntimePaddingShape(pk_node, 0), in LoadInputData()
68 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in CompileGraphImpl() local
69 MS_EXCEPTION_IF_NULL(pk_node); in CompileGraphImpl()
70 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in CompileGraphImpl()
72 if (AnfAlgo::IsParameterWeight(pk_node)) { in CompileGraphImpl()
73 const auto &param_value = pk_node->default_param(); in CompileGraphImpl()
[all …]
Dgpu_inference_session.cc42 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in LoadInputData() local
43 MS_EXCEPTION_IF_NULL(pk_node); in LoadInputData()
44 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in LoadInputData()
46 if (!AnfAlgo::IsParameterWeight(pk_node)) { in LoadInputData()
48 if (!device_address->SyncHostToDevice(trans::GetRuntimePaddingShape(pk_node, 0), in LoadInputData()
68 auto pk_node = input_nodes[i]->cast<ParameterPtr>(); in CompileGraphImpl() local
69 MS_EXCEPTION_IF_NULL(pk_node); in CompileGraphImpl()
70 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in CompileGraphImpl()
72 if (AnfAlgo::IsParameterWeight(pk_node)) { in CompileGraphImpl()
73 const auto &param_value = pk_node->default_param(); in CompileGraphImpl()
[all …]
Dgpu_session.cc341 auto pk_node = input_node->cast<ParameterPtr>(); in LoadInputData() local
342 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in LoadInputData()
354 AnfAlgo::SetOutputAddr(tensor_address, 0, pk_node.get()); in LoadInputData()
360 if (AnfAlgo::IsParameterWeight(pk_node) || UpdatedByAssign(kernel_graph, input_node) || in LoadInputData()
365 …if (!device_address->SyncHostToDevice(trans::GetRuntimePaddingShape(pk_node, 0), size, tensor->dat… in LoadInputData()
369 if (kernel_graph->IsUpdatedParameter(pk_node)) { in LoadInputData()
Dsession_basic.cc2630 auto pk_node = input_node->cast<ParameterPtr>(); in CheckPSModeConsistence() local
2631 MS_EXCEPTION_IF_NULL(pk_node); in CheckPSModeConsistence()
2632 auto param_info_ptr = pk_node->param_info(); in CheckPSModeConsistence()
2633 const std::string &param_name = pk_node->fullname_with_scope(); in CheckPSModeConsistence()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc743 auto pk_node = input_node->cast<ParameterPtr>(); in StepLoadCtrlInputs() local
744 MS_EXCEPTION_IF_NULL(pk_node); in StepLoadCtrlInputs()
745 if (tensor->NeedSyncHostToDevice() || !pk_node->has_default()) { in StepLoadCtrlInputs()
750 auto pk_node = input_node->cast<ParameterPtr>(); in StepLoadCtrlInputs() local
751 MS_EXCEPTION_IF_NULL(pk_node); in StepLoadCtrlInputs()
752 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in StepLoadCtrlInputs()
755 if (!device_address->SyncHostToDevice(trans::GetRuntimePaddingShape(pk_node, 0), in StepLoadCtrlInputs()
/third_party/mindspore/mindspore/ccsrc/ps/
Dworker.cc244 auto pk_node = input_node->cast<ParameterPtr>(); in InitPSParamAndOptim() local
245 MS_EXCEPTION_IF_NULL(pk_node); in InitPSParamAndOptim()
246 const std::string &param_name = pk_node->fullname_with_scope(); in InitPSParamAndOptim()
256 auto param_info_ptr = pk_node->param_info(); in InitPSParamAndOptim()