| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | pserver_kernel.h | 35 …virtual void InitKernel(const std::shared_ptr<std::vector<std::shared_ptr<std::vector<size_t>>>> &… in InitKernel() function
36 virtual void InitKernel(const CNodePtr &cnode, in InitKernel() function
|
| D | push_kernel.h | 79 void InitKernel(const CNodePtr &kernel_node) { return; } in InitKernel() function
|
| D | pull_kernel.h | 72 void InitKernel(const CNodePtr &kernel_node) { return; } in InitKernel() function
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | apply_momentum_kernel.h | 38 void InitKernel(const CNodePtr &cnode) override { in InitKernel() function
|
| D | sgd_kernel.h | 38 void InitKernel(const CNodePtr &cnode) override { in InitKernel() function
|
| D | dense_grad_accum_kernel.h | 39 void InitKernel(const CNodePtr &kernel_node) override { in InitKernel() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | debug_cpu_kernel.cc | 28 void DebugCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::DebugCPUKernel
|
| D | l2loss_cpu_kernel.cc | 28 void L2LossCPUKernel<T>::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::L2LossCPUKernel
|
| D | reshape_cpu_kernel.cc | 27 void ReshapeCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::ReshapeCPUKernel
|
| D | allgather_cpu_kernel.cc | 30 void AllGatherCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::AllGatherCPUKernel
|
| D | equal_count_cpu_kernel.cc | 27 void EqualCountCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::EqualCountCPUKernel
|
| D | apply_momentum_cpu_kernel.cc | 28 void ApplyMomentumCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::ApplyMomentumCPUKernel
|
| D | expm1_cpu_kernel.cc | 28 void Expm1CPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::Expm1CPUKernel
|
| D | select_cpu_kernel.cc | 28 void SelectCPUKernel<T>::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::SelectCPUKernel
|
| D | reduce_scatter_cpu_kernel.cc | 32 void ReduceScatterCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::ReduceScatterCPUKernel
|
| D | smooth_l1_loss_grad_cpu_kernel.cc | 28 void SmoothL1LossGradCPUKernel<T>::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::SmoothL1LossGradCPUKernel
|
| D | broadcast_to_cpu_kernel.cc | 28 void BroadcastToCPUKernel<T>::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::BroadcastToCPUKernel
|
| D | unsorted_segment_sum_cpu_kernel.cc | 28 void UnsortedSegmentSumCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::UnsortedSegmentSumCPUKernel
|
| D | print_cpu_kernel.cc | 27 void PrintCPUKernel<T>::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::PrintCPUKernel
|
| D | masked_select_grad_cpu_kernel.cc | 27 void MaskedSelectGradCPUKernel<T>::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::MaskedSelectGradCPUKernel
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | log_softmax_cpu_kernel.cc | 30 void LogSoftmaxCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::LogSoftmaxCPUKernel
|
| D | log_softmax_grad_cpu_kernel.cc | 30 void LogSoftmaxGradCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::LogSoftmaxGradCPUKernel
|
| D | assignadd_cpu_kernel.cc | 29 void AssignAddCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::AssignAddCPUKernel
|
| D | softmax_cpu_kernel.cc | 30 void SoftmaxCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::SoftmaxCPUKernel
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/ |
| D | round_kernel.h | 55 void InitKernel(const CNodePtr &kernel_node) override {} in InitKernel() function
|