Home
last modified time | relevance | path

Searched refs:MS_CTX_GRAD_FOR_SCALAR (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/pybind_api/utils/
Dms_context_py.cc100 .value("grad_for_scalar", MsCtxParam::MS_CTX_GRAD_FOR_SCALAR) in __anon41857d0f0202()
/third_party/mindspore/mindspore/core/utils/
Dms_context.cc93 set_param<bool>(MS_CTX_GRAD_FOR_SCALAR, false); in MsContext()
Dms_context.h89 MS_CTX_GRAD_FOR_SCALAR, enumerator
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.cc411 …(MsContext::GetInstance()->get_param<bool>(MS_CTX_GRAD_FOR_SCALAR) && (*sequeue)[1]->BuildType() !… in CheckTailGradFristSequence()
449 …(MsContext::GetInstance()->get_param<bool>(MS_CTX_GRAD_FOR_SCALAR) && (*sequeue)[i]->BuildType() !… in GenerateSequeueFuncGraph()
/third_party/mindspore/mindspore/core/abstract/
Dprim_others.cc184 if (!MsContext::GetInstance()->get_param<bool>(MS_CTX_GRAD_FOR_SCALAR)) { in InferImplDepend()
Dabstract_value.cc161 if (context->get_param<bool>(MS_CTX_GRAD_FOR_SCALAR)) { in Broaden()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Daction.cc73 …(MsContext::GetInstance()->get_param<bool>(MS_CTX_GRAD_FOR_SCALAR) && par_abs->BuildType() != null… in UpdateFuncGraphParameter()
Dpipeline.cc129 … (MsContext::GetInstance()->get_param<bool>(MS_CTX_GRAD_FOR_SCALAR) && value->isa<Scalar>()); in ArgsToAbstract()