/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() 44 … const CNodePtr &cnode_ptr, const std::vector<void *> &runtime_args, const std::string &ori_key) in AiCoreDynamicKernel()
|
D | host_dynamic_kernel.h | 27 HostDynamicKernel(void *stream, const CNodePtr &cnode_ptr) : DynamicKernel(stream, cnode_ptr) {} in HostDynamicKernel()
|
D | hccl_dynamic_kernel.h | 32 int32_t op_type, int32_t root, void *stream, const CNodePtr &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()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
D | dynamic_broadcast_gradient_args_kernel.h | 29 …DynamicBroadcastGradientArgsKernel(void *stream, const CNodePtr &cnode_ptr) : HostDynamicKernel(st… in DynamicBroadcastGradientArgsKernel()
|
D | dynamic_shape_kernel.h | 28 …DynamicShapeKernel(void *stream, const CNodePtr &cnode_ptr) : HostDynamicKernel(stream, cnode_ptr)… in DynamicShapeKernel()
|
D | dynamic_shape_kernel.cc | 52 device::DynamicKernelPtr DynamicShapeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *s… in GenDynamicKernel()
|
D | dynamic_broadcast_gradient_args_kernel.cc | 213 …DynamicKernelPtr DynamicBroadcastGradientArgsKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, in GenDynamicKernel()
|
/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()
|
D | tbe_kernel_broadcast_selecter.h | 30 explicit TbeKernelBroadCastSelecter(CNodePtr cnode_ptr) : cnode_ptr_(std::move(cnode_ptr)) {} in TbeKernelBroadCastSelecter()
|
/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()
|
D | memcpy_rts_dynamic_kernel.h | 27 …MemcpyRtsDynamicKernel(void *stream, const CNodePtr &cnode_ptr, void *dst, uint32_t dest_max, void… in MemcpyRtsDynamicKernel()
|
/third_party/mindspore/mindspore/ccsrc/runtime/device/executor/ |
D | dynamic_kernel.h | 32 DynamicKernel(void *stream, const CNodePtr &cnode_ptr) in DynamicKernel()
|
/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()
|
D | memcpy_async.cc | 135 device::DynamicKernelPtr MemCpyAsyncKernel::GenDynamicKernel(const CNodePtr &cnode_ptr, void *strea… 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()
|
/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()
|
/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() 89 …void InitDynamicKernel(const CNodePtr &cnode_ptr) { dynamic_kernel_ = std::make_shared<GpuDynamicK… in InitDynamicKernel()
|
/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()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
D | hccl_kernel.cc | 275 device::DynamicKernelPtr HcclKernel::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr) { in GenDynamicKernel()
|
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
D | optimizer_unify_output.cc | 37 auto cnode_ptr = node->cast<CNodePtr>(); in ProcessOutput() local
|
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
D | adam_fusion.cc | 79 auto cnode_ptr = (user.first)->cast<CNodePtr>(); in RelpaceOutputEdge() local
|
D | adam_weight_decay_fusion.cc | 78 auto cnode_ptr = (user.first)->cast<CNodePtr>(); in ReplaceOutputEdge() local
|
/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()
|
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
D | parameter_manager.cc | 57 auto cnode_ptr = cnode->cast<CNodePtr>(); in FindRefKeyNodeUsers() local 136 auto cnode_ptr = cnode->cast<CNodePtr>(); in CNodeWithRefKeys() local
|