Searched defs:NeedGrad (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ | ||
| D | pipeline_transformer.cc | 306 bool PipelineTransformer::NeedGrad(const CNodePtr &cnode) { in NeedGrad() function in mindspore::parallel::PipelineTransformer |
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/ | ||
| D | pynative_utils.cc | 2151 bool AutoGrad::NeedGrad(const std::vector<ValuePtr> &input_values) { in NeedGrad() function in mindspore::pynative::PyNativeAlgo::AutoGrad |