Home
last modified time | relevance | path

Searched defs:cnode_ptr (Results 1 – 25 of 29) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/
Dai_core_dynamic_kernel.h36 … void *stream, const CNodePtr &cnode_ptr, const std::vector<void *> &runtime_args) in AiCoreDynamicKernel()
44const CNodePtr &cnode_ptr, const std::vector<void *> &runtime_args, const std::string &ori_key) in AiCoreDynamicKernel()
Dhost_dynamic_kernel.h27 HostDynamicKernel(void *stream, const CNodePtr &cnode_ptr) : DynamicKernel(stream, cnode_ptr) {} in HostDynamicKernel()
Dhccl_dynamic_kernel.h32 int32_t op_type, int32_t root, void *stream, const CNodePtr &cnode_ptr) in HcclDynamicKernel()
Dai_cpu_dynamic_kernel.h31 …AiCpuDynamicKernel(void *stream, const CNodePtr &cnode_ptr, const std::string &args, const std::st… in AiCpuDynamicKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Ddynamic_broadcast_gradient_args_kernel.h29 …DynamicBroadcastGradientArgsKernel(void *stream, const CNodePtr &cnode_ptr) : HostDynamicKernel(st… in DynamicBroadcastGradientArgsKernel()
Ddynamic_shape_kernel.h28 …DynamicShapeKernel(void *stream, const CNodePtr &cnode_ptr) : HostDynamicKernel(stream, cnode_ptr)… in DynamicShapeKernel()
Ddynamic_shape_kernel.cc52 device::DynamicKernelPtr DynamicShapeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *s… in GenDynamicKernel()
Ddynamic_broadcast_gradient_args_kernel.cc213 …DynamicKernelPtr DynamicBroadcastGradientArgsKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, in GenDynamicKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_reduce_selecter.h28 explicit TbeKernelReduceSelecter(CNodePtr cnode_ptr) : cnode_ptr_(std::move(cnode_ptr)) {} in TbeKernelReduceSelecter()
Dtbe_kernel_broadcast_selecter.h30 explicit TbeKernelBroadCastSelecter(CNodePtr cnode_ptr) : cnode_ptr_(std::move(cnode_ptr)) {} in TbeKernelBroadCastSelecter()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/rts/
Dprofiling_rts_dynamic_kernel.h27 …ProfilingRtsDynamicKernel(void *stream, const CNodePtr &cnode_ptr, uint64_t log_id, bool notify, u… in ProfilingRtsDynamicKernel()
Dmemcpy_rts_dynamic_kernel.h27 …MemcpyRtsDynamicKernel(void *stream, const CNodePtr &cnode_ptr, void *dst, uint32_t dest_max, void… in MemcpyRtsDynamicKernel()
/third_party/mindspore/mindspore/ccsrc/runtime/device/executor/
Ddynamic_kernel.h32 DynamicKernel(void *stream, const CNodePtr &cnode_ptr) in DynamicKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Dprofiling_kernel_mod.cc68 device::DynamicKernelPtr ProfilingKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stre… in GenDynamicKernel()
Dmemcpy_async.cc135 device::DynamicKernelPtr MemCpyAsyncKernel::GenDynamicKernel(const CNodePtr &cnode_ptr, void *strea… in GenDynamicKernel()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/
Dallreduce_node.cc66 Status AllreduceNode::Init(const CNodePtr &cnode_ptr) { in Init()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconvert_tuple_input_to_dynamic_input.cc59 void ConvertMakeTupleInputToPlantInputs(const FuncGraphPtr &graph, const CNodePtr &cnode_ptr) { in ConvertMakeTupleInputToPlantInputs()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/
Dgpu_kernel.h71 explicit GpuDynamicKernel(const CNodePtr &cnode_ptr) : DynamicKernel(nullptr, cnode_ptr) {} in GpuDynamicKernel()
89 …void InitDynamicKernel(const CNodePtr &cnode_ptr) { dynamic_kernel_ = std::make_shared<GpuDynamicK… in InitDynamicKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_mod.cc206 device::DynamicKernelPtr AicpuOpKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream… in GenDynamicKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhccl_kernel.cc275 device::DynamicKernelPtr HcclKernel::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) { in GenDynamicKernel()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Doptimizer_unify_output.cc37 auto cnode_ptr = node->cast<CNodePtr>(); in ProcessOutput() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dadam_fusion.cc79 auto cnode_ptr = (user.first)->cast<CNodePtr>(); in RelpaceOutputEdge() local
Dadam_weight_decay_fusion.cc78 auto cnode_ptr = (user.first)->cast<CNodePtr>(); in ReplaceOutputEdge() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.cc113 device::DynamicKernelPtr TbeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr… in GenDynamicKernel()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc57 auto cnode_ptr = cnode->cast<CNodePtr>(); in FindRefKeyNodeUsers() local
136 auto cnode_ptr = cnode->cast<CNodePtr>(); in CNodeWithRefKeys() local

12