Searched defs:grad_executor (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/jit/ |
| D | jit_grad.cc | 238 …d Jit::GetInputArgsNode(const FrontendOpRunInfoPtr &op_run_info, const GradExecutor *grad_executor, in GetInputArgsNode()
253 void Jit::GetWeightsNode(const FrontendOpRunInfoPtr &op_run_info, const GradExecutor *grad_executor, in GetWeightsNode()
287 …id Jit::MakeCNodeForJit(const FrontendOpRunInfoPtr &op_run_info, const GradExecutor *grad_executor, in MakeCNodeForJit()
304 … Jit::MakeAdjointForJit(const FrontendOpRunInfoPtr &op_run_info, const GradExecutor *grad_executor, in MakeAdjointForJit()
355 void Jit::KPynativeWithFProp(const GradExecutor *grad_executor, const autograd::AutoGradPtr &auto_g… in KPynativeWithFProp()
364 …ecordForwardGraphForJit(const FrontendOpRunInfoPtr &op_run_info, const GradExecutor *grad_executor, in RecordForwardGraphForJit()
377 void Jit::GradJitInner(const FrontendOpRunInfoPtr &op_run_info, const GradExecutor *grad_executor, in GradJitInner()
580 const auto &grad_executor = PyNativeAlgo::Common::GetPyNativeExecutor()->grad_executor(); in GradJit() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.h | 48 static inline const GradExecutorPtr &grad_executor() { in grad_executor() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/forward/ |
| D | forward.h | 67 …void set_grad_executor(const GradExecutorPtr &grad_executor) { grad_executor_ = GradExecutorWeakPt… in set_grad_executor()
|
| D | forward.cc | 269 auto grad_executor = grad_executor_.lock(); in grad() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/ |
| D | dynamic_shape.cc | 567 const auto &grad_executor = PyNativeAlgo::Common::GetPyNativeExecutor()->grad_executor(); in TryChangeTopCellToUnknownShape() local
654 const auto &grad_executor = PyNativeAlgo::Common::GetPyNativeExecutor()->grad_executor(); in UpdateArgsAbsToUnknownShapeAbs() local
|