Home
last modified time | relevance | path

Searched defs:abs_func (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/debug/
Danf_dump_utils.cc53 const auto &abs_func = abs->cast<abstract::AbstractFunctionPtr>(); in GetNodeFuncStr() local
/third_party/mindspore/mindspore-src/source/tests/st/dyn_shape_dev/
Dtest_abs_op_dyn_shape.py59 def abs_func(x): function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Drecompute_prepare.h104 auto abs_func = abs->cast<abstract::AbstractFunctionPtr>(); in GetRealOutputNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/
Dutils.cc353 auto abs_func = abs_partial_closure->fn(); in GetFuncGraphFromAbs() local
380 auto abs_func = abs->cast<abstract::AbstractFunctionPtr>(); in GetFuncGraphsFromCallNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc182 const auto &abs_func = abs->cast<abstract::AbstractFunctionPtr>(); in CallGradNodes() local
240 const auto &abs_func = abs->cast<abstract::AbstractFunctionPtr>(); in SetParameterStartForCellShare() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_build/
Dfunc_graph_builder.cc783 auto abs_func = abs->cast<abstract::FuncGraphAbstractClosurePtr>(); in TryToAddNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dfallback.cc823 auto abs_func = abs->cast<abstract::FuncGraphAbstractClosurePtr>(); in GetPyObjForFuncGraphAbstractClosure() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/
Dprogram_specialize.cc82 auto abs_func = node->abstract()->cast_ptr<FuncGraphAbstractClosure>(); in CanSpecializeValueNode() local