| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | apply_momentum_kernel.h | 38 void InitKernel(const CNodePtr &cnode) override { in InitKernel() function
40 ApplyMomentumCPUKernel::InitKernel(cnode); in InitKernel()
|
| D | sgd_kernel.h | 38 void InitKernel(const CNodePtr &cnode) override { in InitKernel() function
39 SGDCPUKernel<T>::InitKernel(cnode); in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/ |
| D | round_kernel.h | 55 void InitKernel(const CNodePtr &kernel_node) override {} in InitKernel() function
59 virtual void InitKernel(size_t threshold_count) = 0;
|
| /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 | embedding_look_up_proxy_kernel.cc | 29 void EmbeddingLookUpProxyKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::ps::EmbeddingLookUpProxyKernel
31 EmbeddingLookUpCPUKernel::InitKernel(kernel_node); in InitKernel()
|
| D | embedding_look_up_proxy_kernel.h | 32 void InitKernel(const CNodePtr &kernel_node) override;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | equal_count_cpu_kernel.h | 32 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | debug_cpu_kernel.h | 33 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | bias_add_grad_cpu_kernel.h | 33 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | one_hot_cpu_kernel.h | 32 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | allgather_cpu_kernel.h | 33 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | embedding_look_up_comm_grad_cpu_kernel.h | 32 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | reduce_scatter_cpu_kernel.h | 32 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | bias_add_cpu_kernel.h | 33 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | multinomial_cpu_kernel.h | 33 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | l2loss_cpu_kernel.h | 34 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | range_cpu_kernel.h | 32 void InitKernel(const CNodePtr &kernel_node) override;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | conv_cpu_kernel.h | 31 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | softmax_cpu_kernel.h | 31 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | log_softmax_cpu_kernel.h | 31 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | log_softmax_grad_cpu_kernel.h | 31 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | conv2d_grad_input_cpu_kernel.h | 31 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | assignadd_cpu_kernel.h | 30 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | conv2d_grad_filter_cpu_kernel.h | 31 void InitKernel(const CNodePtr &kernel_node) override;
|
| D | pooling_avg_grad_cpu_kernel.h | 32 void InitKernel(const CNodePtr &kernel_node) override;
|