Searched defs:origin_node (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | unsorted_segment_sum_fission.cc | 26 CNodePtr CreatePadding(const FuncGraphPtr &graph, const CNodePtr &origin_node, const size_t &pad_di… in CreatePadding()
42 CNodePtr CreateUnsortedSegmentSum(const FuncGraphPtr &graph, const CNodePtr &origin_node, const CNo… in CreateUnsortedSegmentSum()
79 bool CheckInputs(const CNodePtr &origin_node) { in CheckInputs()
110 auto origin_node = node->cast<CNodePtr>(); in Process() local
|
| D | gather_v2_ds_fission.cc | 32 CNodePtr CreatePad(const FuncGraphPtr &graph, const CNodePtr &origin_node, const size_t &pad_dim_si… in CreatePad()
86 CNodePtr CreateGatherV2Ds(const FuncGraphPtr &graph, const CNodePtr &origin_node, const CNodePtr &p… in CreateGatherV2Ds()
129 bool CheckInputs(const CNodePtr &origin_node) { in CheckInputs()
160 auto origin_node = node->cast<CNodePtr>(); in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | grad_var_prepare.cc | 32 static AnfNodePtr GenerateUnpackGraphNode(const AnfNodePtr &origin_node, std::vector<AnfNodePtr> in… in GenerateUnpackGraphNode()
|
| D | incorporate_getitem.h | 460 AnfNodePtr TransformFuncGraph(const FuncGraphManagerPtr &manager, const AnfNodePtr &origin_node) { in TransformFuncGraph()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | recompute.cc | 326 CNodePtr CreateNewRecomputedNode(const FuncGraphPtr &graph, const CNodePtr &origin_node, in CreateNewRecomputedNode()
337 CNodePtr NewRecomputedNode(const FuncGraphPtr &graph, const CNodePtr &origin_node, in NewRecomputedNode()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_bn_fusion.cc | 112 STATUS CalEstimatedData(const AnfNodePtr &origin_node, const AnfNodePtr &scale_factor_node) { in CalEstimatedData()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | program_specialize.cc | 387 const auto &origin_node = conf_iter->first->node(); in BuildReplacedNode() local
840 AnfNodePtr FuncGraphSpecializer::BuildPossibleValueNode(const AnfNodePtr &origin_node, const Abstra… in BuildPossibleValueNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | kprim.cc | 170 AnfNodePtr GetPythonOps(const FuncGraphPtr &fg, const AnfNodePtr &origin_node, const PrimitivePtr &… in GetPythonOps()
|