| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | convert_tuple_input_to_dynamic_input.cc | 59 void ConvertMakeTupleInputToPlantInputs(const FuncGraphPtr &graph, const CNodePtr &cnode_ptr) { in ConvertMakeTupleInputToPlantInputs() argument
60 MS_EXCEPTION_IF_NULL(cnode_ptr); in ConvertMakeTupleInputToPlantInputs()
62 if (AnfAlgo::CheckPrimitiveType(cnode_ptr, prim::kPrimCall) || in ConvertMakeTupleInputToPlantInputs()
63 AnfAlgo::CheckPrimitiveType(cnode_ptr, prim::kPrimPartial)) { in ConvertMakeTupleInputToPlantInputs()
68 plant_inputs.push_back(AnfAlgo::GetCNodePrimitiveNode(cnode_ptr)); in ConvertMakeTupleInputToPlantInputs()
69 size_t input_num = cnode_ptr->inputs().size() - 1; in ConvertMakeTupleInputToPlantInputs()
71 auto input_node = AnfAlgo::GetInputNode(cnode_ptr, i); in ConvertMakeTupleInputToPlantInputs()
82 AnfAlgo::SetNodeAttr(kAttrDynInputSizes, MakeValue(dyn_input_sizes), cnode_ptr); in ConvertMakeTupleInputToPlantInputs()
83 cnode_ptr->set_inputs(plant_inputs); in ConvertMakeTupleInputToPlantInputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | optimizer_unify_output.cc | 37 auto cnode_ptr = node->cast<CNodePtr>(); in ProcessOutput() local
38 MS_EXCEPTION_IF_NULL(cnode_ptr); in ProcessOutput()
40 auto abstract = cnode_ptr->abstract(); in ProcessOutput()
43 …if (AnfAlgo::HasNodeAttr("optim_output_passed", cnode_ptr) && abstract->isa<abstract::AbstractTupl… in ProcessOutput()
46 AnfAlgo::SetNodeAttr("optim_output_passed", MakeValue(true), cnode_ptr); in ProcessOutput()
53 cnode_ptr->set_abstract(abstract_tuple); in ProcessOutput()
56 auto get_item = graph->NewCNode({NewValueNode(prim::kPrimTupleGetItem), cnode_ptr, index}); in ProcessOutput()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | ai_core_dynamic_kernel.h | 36 … void *stream, const CNodePtr &cnode_ptr, const std::vector<void *> &runtime_args) in AiCoreDynamicKernel() argument
37 : DynamicKernel(stream, cnode_ptr), in AiCoreDynamicKernel()
44 … const CNodePtr &cnode_ptr, const std::vector<void *> &runtime_args, const std::string &ori_key) in AiCoreDynamicKernel() argument
45 : DynamicKernel(stream, cnode_ptr), in AiCoreDynamicKernel()
|
| D | host_dynamic_kernel.h | 27 HostDynamicKernel(void *stream, const CNodePtr &cnode_ptr) : DynamicKernel(stream, cnode_ptr) {} in HostDynamicKernel() argument
|
| D | hccl_dynamic_kernel.h | 32 int32_t op_type, int32_t root, void *stream, const CNodePtr &cnode_ptr) in HcclDynamicKernel() argument
33 : DynamicKernel(stream, cnode_ptr), in HcclDynamicKernel()
|
| D | ai_cpu_dynamic_kernel.h | 31 …AiCpuDynamicKernel(void *stream, const CNodePtr &cnode_ptr, const std::string &args, const std::st… in AiCpuDynamicKernel() argument
33 : DynamicKernel(stream, cnode_ptr), in AiCpuDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | dynamic_shape_kernel.h | 28 …DynamicShapeKernel(void *stream, const CNodePtr &cnode_ptr) : HostDynamicKernel(stream, cnode_ptr)… in DynamicShapeKernel() argument
37 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
|
| D | dynamic_broadcast_gradient_args_kernel.h | 29 …astGradientArgsKernel(void *stream, const CNodePtr &cnode_ptr) : HostDynamicKernel(stream, cnode_p… in DynamicBroadcastGradientArgsKernel() argument
38 device::DynamicKernelPtr GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) override;
|
| D | dynamic_shape_kernel.cc | 52 device::DynamicKernelPtr DynamicShapeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *s… in GenDynamicKernel() argument
53 return std::make_shared<DynamicShapeKernel>(stream_ptr, cnode_ptr); in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/core/load_mindir/ |
| D | anf_model_parser.cc | 862 void MSANFModelParser::SetCNodeAbastract(const mind_ir::NodeProto &node_proto, CNodePtr cnode_ptr) { in SetCNodeAbastract() argument
865 auto operatorPtr = cnode_ptr->input(0); in SetCNodeAbastract()
871 cnode_ptr->set_abstract(nullptr); in SetCNodeAbastract()
878 cnode_ptr->set_abstract(nullptr); in SetCNodeAbastract()
898 cnode_ptr->set_abstract(kUMonad->ToAbstract()); in SetCNodeAbastract()
900 cnode_ptr->set_abstract(kBool->ToAbstract()); in SetCNodeAbastract()
903 for (size_t index = 1; index < cnode_ptr->inputs().size(); ++index) { in SetCNodeAbastract()
904 auto abs = cnode_ptr->input(index)->abstract(); in SetCNodeAbastract()
913 cnode_ptr->set_abstract(std::make_shared<abstract::AbstractTuple>(elem)); in SetCNodeAbastract()
920 cnode_ptr->set_abstract(iter->second); in SetCNodeAbastract()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_mod.cc | 113 device::DynamicKernelPtr TbeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr… in GenDynamicKernel() argument
117 …device::KernelRuntime::GenLaunchArgs(*this, cnode_ptr, &kernel_inputs, &kernel_workspaces, &kernel… in GenDynamicKernel()
118 auto dynamic_flag = AnfAlgo::IsDynamicShape(cnode_ptr); in GenDynamicKernel()
155 … stream_ptr, cnode_ptr, runtime_args, origin_key); in GenDynamicKernel()
162 … op_para_size, stream_ptr, cnode_ptr, runtime_args); in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/ |
| D | allreduce_node.cc | 66 Status AllreduceNode::Init(const CNodePtr &cnode_ptr) { in Init() argument
67 if (cnode_ptr == nullptr) { in Init()
71 cnode_ptr_ = cnode_ptr; in Init()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/rts/ |
| D | profiling_rts_dynamic_kernel.h | 27 …ProfilingRtsDynamicKernel(void *stream, const CNodePtr &cnode_ptr, uint64_t log_id, bool notify, u… in ProfilingRtsDynamicKernel() argument
28 : DynamicKernel(stream, cnode_ptr), log_id_(log_id), notify_(notify), flags_(flags) {} in ProfilingRtsDynamicKernel()
|
| D | memcpy_rts_dynamic_kernel.h | 27 …MemcpyRtsDynamicKernel(void *stream, const CNodePtr &cnode_ptr, void *dst, uint32_t dest_max, void… in MemcpyRtsDynamicKernel() argument
29 … : DynamicKernel(stream, cnode_ptr), dst_(dst), dest_max_(dest_max), src_(src), count_(count) {} in MemcpyRtsDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/executor/ |
| D | dynamic_kernel.h | 32 DynamicKernel(void *stream, const CNodePtr &cnode_ptr) in DynamicKernel() argument
34 cnode_ptr_(cnode_ptr), in DynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_reduce_selecter.h | 28 explicit TbeKernelReduceSelecter(CNodePtr cnode_ptr) : cnode_ptr_(std::move(cnode_ptr)) {} in TbeKernelReduceSelecter() argument
|
| D | tbe_kernel_broadcast_selecter.h | 30 explicit TbeKernelBroadCastSelecter(CNodePtr cnode_ptr) : cnode_ptr_(std::move(cnode_ptr)) {} in TbeKernelBroadCastSelecter() argument
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/ |
| D | profiling_utils.cc | 259 CNodePtr cnode_ptr = graph_ptr->NewCNode(inputs); in CreateProfilingCNode() local
260 MS_EXCEPTION_IF_NULL(cnode_ptr); in CreateProfilingCNode()
261 AnfAlgo::SetSelectKernelBuildInfo(selected_kernel_builder.Build(), cnode_ptr.get()); in CreateProfilingCNode()
262 cnode_ptr->set_abstract(type_none_abstract); in CreateProfilingCNode()
267 AnfAlgo::SetNodeAttr(ProfilingUtils::kNotify, notify_value, cnode_ptr); in CreateProfilingCNode()
268 AnfAlgo::SetNodeAttr(ProfilingUtils::kProfilerTraceId, trace_id_value, cnode_ptr); in CreateProfilingCNode()
269 AnfAlgo::SetNodeAttr(ProfilingUtils::kFlags, flags_value, cnode_ptr); in CreateProfilingCNode()
270 return NOT_NULL(cnode_ptr); in CreateProfilingCNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | profiling_kernel_mod.cc | 68 device::DynamicKernelPtr ProfilingKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stre… in GenDynamicKernel() argument
69 …return std::make_shared<ProfilingRtsDynamicKernel>(stream_ptr, cnode_ptr, log_id_, notify_, flags_… in GenDynamicKernel()
|
| D | memcpy_async.cc | 135 device::DynamicKernelPtr MemCpyAsyncKernel::GenDynamicKernel(const CNodePtr &cnode_ptr, void *strea… in GenDynamicKernel() argument
139 …device::KernelRuntime::GenLaunchArgs(*this, cnode_ptr, &kernel_inputs, &kernel_workspaces, &kernel… in GenDynamicKernel()
160 return std::make_shared<MemcpyRtsDynamicKernel>(stream_ptr, cnode_ptr, kernel_outputs[0]->addr, in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | adam_fusion.cc | 79 auto cnode_ptr = (user.first)->cast<CNodePtr>(); in RelpaceOutputEdge() local
80 MS_EXCEPTION_IF_NULL(cnode_ptr); in RelpaceOutputEdge()
81 cnode_ptr->set_input(monad_index, u_input); in RelpaceOutputEdge()
82 cnode_ptr->set_input(adam_index, adam); in RelpaceOutputEdge()
|
| D | adam_weight_decay_fusion.cc | 78 auto cnode_ptr = (user.first)->cast<CNodePtr>(); in ReplaceOutputEdge() local
79 MS_EXCEPTION_IF_NULL(cnode_ptr); in ReplaceOutputEdge()
80 cnode_ptr->set_input(monad_index, u_input); in ReplaceOutputEdge()
81 cnode_ptr->set_input(adam_weight_decay_index, adam_weight_decay); in ReplaceOutputEdge()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel.h | 71 explicit GpuDynamicKernel(const CNodePtr &cnode_ptr) : DynamicKernel(nullptr, cnode_ptr) {} in GpuDynamicKernel() argument
89 …void InitDynamicKernel(const CNodePtr &cnode_ptr) { dynamic_kernel_ = std::make_shared<GpuDynamicK… in InitDynamicKernel() argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_mod.cc | 206 device::DynamicKernelPtr AicpuOpKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream… in GenDynamicKernel() argument
210 …device::KernelRuntime::GenLaunchArgs(*this, cnode_ptr, &kernel_inputs, &kernel_workspaces, &kernel… in GenDynamicKernel()
213 …return std::make_shared<AicpuDynamicKernel>(stream_ptr, cnode_ptr, args_, ext_info_, node_so_, nod… in GenDynamicKernel()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.cc | 57 auto cnode_ptr = cnode->cast<CNodePtr>(); in FindRefKeyNodeUsers() local
58 …if ((cnode_ptr == nullptr) || !IsValueNode<Primitive>(cnode_ptr->input(0)) || !IsCareNode(cnode_pt… in FindRefKeyNodeUsers()
136 auto cnode_ptr = cnode->cast<CNodePtr>(); in CNodeWithRefKeys() local
137 auto inputs = cnode_ptr->inputs(); in CNodeWithRefKeys()
|