Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute.h195 void RecordGradOpInfo(const OpExecInfoPtr &op_exec_info, const ValuePtr &op_out);
203 …void UpdateMsFunctionForwardTensors(const OpExecInfoPtr &op_exec_info, const ValuePtr &new_forward…
209 void DoOpGrad(const OpExecInfoPtr &op_exec_info, const CNodePtr &cnode, const ValuePtr &op_out);
211 …void UpdateForwardTensorInfoInBpropGraph(const OpExecInfoPtr &op_exec_info, const ValuePtr &op_out…
318 std::function<void(py::object *, const OpExecInfoPtr &)> RunOpS = [this](auto &&PH1, auto &&PH2) {
322 void RunOpInner(py::object *ret, const OpExecInfoPtr &op_exec_info);
323 OpExecInfoPtr GenerateOpExecInfo(const py::args &args);
327 CNodePtr ConstructForwardGraph(const OpExecInfoPtr &op_exec_info);
332 MsBackendPolicy InitEnv(const OpExecInfoPtr &op_exec_info);
333 py::tuple RunOpWithInitBackendPolicy(const OpExecInfoPtr &op_exec_info);
[all …]
Dpynative_execute_ge.h39 bool BuildSingleOpGraph(const OpExecInfoPtr &op_exec_info, const std::vector<GeTensorPtr> &inputs,
42 py::object RunOpInGE(const OpExecInfoPtr &op_exec_info, PynativeStatusCode *status);
Dpynative_execute_ge.cc77 bool SetInputsForSingleOpGraph(const OpExecInfoPtr &op_exec_info, const std::vector<GeTensorPtr> &i… in SetInputsForSingleOpGraph()
121 bool BuildSingleOpGraph(const OpExecInfoPtr &op_exec_info, const std::vector<GeTensorPtr> &inputs, in BuildSingleOpGraph()
170 void ToTensorPtr(const OpExecInfoPtr op_exec_info, std::vector<GeTensorPtr> *const inputs) { in ToTensorPtr()
190 PynativeStatusCode ConvertAttributes(const OpExecInfoPtr &op_exec_info, const std::vector<GeTensorP… in ConvertAttributes()
223 std::vector<MeTensorPtr> ConvertOutputTensors(const OpExecInfoPtr &op_exec_info, in ConvertOutputTensors()
264 py::object RunOpInGE(const OpExecInfoPtr &op_exec_info, PynativeStatusCode *status) { in RunOpInGE()
Dpynative_execute.cc308 void GetSingleOpGraphInfo(const OpExecInfoPtr &op_exec_info, const std::vector<tensor::TensorPtr> &… in GetSingleOpGraphInfo()
497 void ConstructInputTensor(const OpExecInfoPtr &op_run_info, std::vector<int64_t> *tensors_mask, in ConstructInputTensor()
546 void ConvertAttrToUnifyMindIR(const OpExecInfoPtr &op_run_info) { in ConvertAttrToUnifyMindIR()
667 void ReplaceNewTensorsInGradGraph(const TopCellInfoPtr &top_cell, const OpExecInfoPtr &op_exec_info, in ReplaceNewTensorsInGradGraph()
800 OpExecInfoPtr op_exec_info = executor->forward_executor()->GenerateOpExecInfo(args); in RealRunOp()
875 void ForwardExecutor::RunOpInner(py::object *ret, const OpExecInfoPtr &op_exec_info) { in RunOpInner()
898 OpExecInfoPtr ForwardExecutor::GenerateOpExecInfo(const py::args &args) { in GenerateOpExecInfo()
923 void ForwardExecutor::SetCastForInputs(const OpExecInfoPtr &op_exec_info) { in SetCastForInputs()
936 void ForwardExecutor::RunMixedPrecisionCastOp(const OpExecInfoPtr &op_exec_info, py::object *ret) { in RunMixedPrecisionCastOp()
964 void ForwardExecutor::GetInputsArgsSpec(const OpExecInfoPtr &op_exec_info, in GetInputsArgsSpec()
[all …]
Dbase.h68 using OpExecInfoPtr = std::shared_ptr<OpExecInfo>; variable
/third_party/mindspore/tests/ut/cpp/pynative/
Dpynative_execute_test.cc47 OpExecInfoPtr ConstructOpExecInfo() { in ConstructOpExecInfo()