Home
last modified time | relevance | path

Searched defs:CNodePtr (Results 1 – 25 of 36) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dconvert_pad_v3_paddings.h42 …virtual const AnfNodePtr CreateDynPaddingsNode(const FuncGraphPtr &, const CNodePtr &) const { ret… in CreateDynPaddingsNode()
54 virtual bool ExpandInputXDims(const FuncGraphPtr &, const CNodePtr &) const { return false; } in ExpandInputXDims()
55 virtual void ReduceOutputDims(const FuncGraphPtr &, const CNodePtr &) const {} in ReduceOutputDims()
82 bool ExpandInputXDims(const FuncGraphPtr &, const CNodePtr &) const override { return false; } in ExpandInputXDims()
83 void ReduceOutputDims(const FuncGraphPtr &, const CNodePtr &) const override {} in ReduceOutputDims()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/checker/
Dcustom_op_checker.cc23 bool CustomOpChecker::Check(api::CNodePtr, int32_t output_num, mindspore::Format) { return true; } in Check()
Dresize_checker.cc77 bool IsDoubleResize(const api::CNodePtr &, const ShapeVector &input_shape, const ShapeVector &outpu… in IsDoubleResize()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dinplace_assign_builder.h64 virtual void SetTargetAttrs(const CNodePtr &) {} in SetTargetAttrs()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dpass_utils.cc96 std::unordered_map<CNodePtr, CNodePtr> *backward_matmul_dx_dw_map) { in ExtractBackwardMatMul()
125 std::unordered_map<CNodePtr, CNodePtr> *backward_matmul_dx_dw_map) { in ExtendDxDwMap()
Doverlap_grad_comm.cc96 … const std::unordered_map<CNodePtr, CNodePtr> &backward_matmul_dx_dw_map, in InsertDependForDxAndGradComm()
185 const std::unordered_map<CNodePtr, CNodePtr> &backward_matmul_dx_dw_map) { in ExtractDxGradCommMap()
Dbias_add_comm_swap.cc100 void FindAllValidAddNode(const FuncGraphPtr &graph, HashMap<CNodePtr, AnfNodePtr> *add_node_map) { in FindAllValidAddNode()
229 void HandleAddNode(HashMap<CNodePtr, AnfNodePtr> *add_node_map) { in HandleAddNode()
Dassign_add_opt.cc43 … const std::unordered_map<CNodePtr, CNodePtr> &backward_matmul_dx_dw_map) { in GetMatmulDwNodeFront()
98 const std::unordered_map<CNodePtr, CNodePtr> &backward_matmul_dx_dw_map) { in InsertConcat()
154 … const std::unordered_map<CNodePtr, CNodePtr> &backward_matmul_dx_dw_map, in MergeMultiMatmulAssignAdd()
Doverlap_gradmatmul_and_gradallreduce.cc55 … const std::unordered_map<CNodePtr, CNodePtr> &backward_matmul_dx_dw_map, in GetCommInputMatMulNode()
114 const std::unordered_map<CNodePtr, CNodePtr> &backward_matmul_dx_dw_map, in OverLapGradMatMul()
Dcomp_comm_scheduling.cc909 std::unordered_map<CNodePtr, TaskPtr> *cnode_to_task_map_ptr) { in InsertTaskGraph()
925 … std::unordered_map<CNodePtr, TaskPtr> *cnode_to_task_map_ptr) { in ExtractSchedulingInput()
995 std::unordered_map<CNodePtr, TaskPtr> *cnode_to_task) { in AddRealDependencies()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/rts/
Drt_kernel_info.h37 virtual std::vector<std::shared_ptr<kernel::KernelBuildInfo>> GetKernelInfo(const CNodePtr &) { in GetKernelInfo()
/third_party/mindspore/mindspore-src/source/mindspore/core/mindapi/ir/
Danf.h105 using CNodePtr = SharedPtr<CNode>; variable
Dcommon.h52 using CNodePtr = SharedPtr<CNode>; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Deliminate_redundant_op.cc73 bool TransOpEliminateCondition(const CNodePtr &, const CNodePtr &) { return true; } in TransOpEliminateCondition()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/debug/rdr/
Dbase_recorder.h51 using CNodePtr = std::shared_ptr<CNode>; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dgraph_kernel_info.h34 virtual void SetKernelInfo(const CNodePtr &, KernelType) {} in SetKernelInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dflops_collection.cc252 void GetBpNodeInfos(std::unordered_map<CNodePtr, OpInfoPtr> *op_info_map_dx_dw_map, in GetBpNodeInfos()
274 std::unordered_map<CNodePtr, OpInfoPtr> bprop_op_info_map_dx_dw_map) { in ToJson()
302 const std::unordered_map<CNodePtr, OpInfoPtr> &bprop_op_info_map_dx_dw_map) { in GetCalOps()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dmatmul_scale_fusion.cc33 PrimitiveCPtr MatMulScaleFusion::BuildNewPrimitive(const CNodePtr &, const CNodePtr &prev_cnode) co… in BuildNewPrimitive() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Drecompute_helper.cc527 mindspore::HashMap<CNodePtr, CNodePtr> *origin_to_recomputed_nodes) { in NewRecomputedNode()
602 mindspore::HashMap<CNodePtr, CNodePtr> *origin_to_new_target_nodes, in DuplicateRecomputedNodes()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/c_api/src/
Dcommon.h74 using CNodePtr = mindspore::CNodePtr; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/quant_helper/
Dpropagate_quant_param_pass.cc243 std::map<CNodePtr, DependNodes> *node_depends) { in FindNodeDepends()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/hcom/
Dinsert_depend_for_all_gather_ge.cc77 const std::unordered_map<CNodePtr, CNodePtr> &allgather_output_another_input) { in InsertDepend()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/
Dkernel_graph.h443 const std::pair<CNodePtr, CNodePtr> &send_recv_pair) { in InsertSendRecvPairForParallelOpInputs()
453 const std::pair<CNodePtr, CNodePtr> &send_recv_pair) { in InsertSendRecvPairForParallelOpOutputs()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Drecompute.cc418 std::unordered_map<CNodePtr, CNodePtr> *origin_to_new_nodes) { in MoveKCallerToBprop()
559 … const CNodePtr &node, mindspore::HashMap<CNodePtr, bool> *is_from_recompute_k_fg_caller) { in IsFromRecomputeKFgCaller()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Ddynamic_rnn_grad_fission_v2.cc810 …fNodePtr DynamicRnnGradFissionV2::CreateValueNode(const FuncGraphPtr &func_graph, const CNodePtr &, in CreateValueNode()
829 …odePtr DynamicRnnGradFissionV2::CreateDbReduceSum(const FuncGraphPtr &func_graph, const CNodePtr &, in CreateDbReduceSum()

12