| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | sgd_kernel.h | 44 bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspace, in Launch() function
|
| D | apply_momentum_kernel.h | 45 bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspace, in Launch() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_all_gather.cc | 22 bool HcomAllGatherKernel::Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPt… in Launch() function in mindspore::kernel::HcomAllGatherKernel
|
| D | hcom_all_reduce_scatter.cc | 22 bool HcomAllReduceScatterKernel::Launch(const std::vector<AddressPtr> &inputs, const std::vector<Ad… in Launch() function in mindspore::kernel::HcomAllReduceScatterKernel
|
| D | hcom_send.cc | 22 bool HcomSendKernel::Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &, in Launch() function in mindspore::kernel::HcomSendKernel
|
| D | hcom_all_broadcast.cc | 23 bool HcomAllBroadCastKernel::Launch(const std::vector<AddressPtr> &inputs, const std::vector<Addres… in Launch() function in mindspore::kernel::HcomAllBroadCastKernel
|
| D | hcom_receive.cc | 22 bool HcomReceiveKernel::Launch(const std::vector<AddressPtr> &, const std::vector<AddressPtr> &, in Launch() function in mindspore::kernel::HcomReceiveKernel
|
| D | hcom_all_reduce.cc | 22 bool HcomAllReduceKernel::Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPt… in Launch() function in mindspore::kernel::HcomAllReduceKernel
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | debug_cpu_kernel.cc | 33 bool DebugCPUKernel::Launch(const std::vector<kernel::AddressPtr> &inputs, const std::vector<kernel… in Launch() function in mindspore::kernel::DebugCPUKernel
|
| D | unique_with_pad_cpu_kernel.cc | 27 bool UniqueWithPadCPUKernel::Launch(const std::vector<kernel::AddressPtr> &inputs, in Launch() function in mindspore::kernel::UniqueWithPadCPUKernel
|
| D | reshape_cpu_kernel.cc | 35 bool ReshapeCPUKernel::Launch(const std::vector<kernel::AddressPtr> &inputs, const std::vector<kern… in Launch() function in mindspore::kernel::ReshapeCPUKernel
|
| D | l2loss_cpu_kernel.cc | 38 bool L2LossCPUKernel<T>::Launch(const std::vector<kernel::AddressPtr> &inputs, const std::vector<ke… in Launch() function in mindspore::kernel::L2LossCPUKernel
|
| D | allgather_cpu_kernel.cc | 38 bool AllGatherCPUKernel::Launch(const std::vector<kernel::AddressPtr> &inputs, const std::vector<ke… in Launch() function in mindspore::kernel::AllGatherCPUKernel
|
| D | apply_momentum_cpu_kernel.cc | 33 bool ApplyMomentumCPUKernel::Launch(const std::vector<kernel::AddressPtr> &inputs, in Launch() function in mindspore::kernel::ApplyMomentumCPUKernel
|
| D | equal_count_cpu_kernel.cc | 32 bool EqualCountCPUKernel::Launch(const std::vector<kernel::AddressPtr> &inputs, const std::vector<k… in Launch() function in mindspore::kernel::EqualCountCPUKernel
|
| D | expm1_cpu_kernel.cc | 37 bool Expm1CPUKernel::Launch(const std::vector<kernel::AddressPtr> &inputs, const std::vector<kernel… in Launch() function in mindspore::kernel::Expm1CPUKernel
|
| D | select_cpu_kernel.cc | 38 bool SelectCPUKernel<T>::Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr… in Launch() function in mindspore::kernel::SelectCPUKernel
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/control/ |
| D | send_gpu_kernel.h | 35 …bool Launch(const std::vector<AddressPtr> &, const std::vector<AddressPtr> &, const std::vector<Ad… in Launch() function
|
| D | recv_gpu_kernel.h | 35 …bool Launch(const std::vector<AddressPtr> &, const std::vector<AddressPtr> &, const std::vector<Ad… in Launch() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | log_softmax_cpu_kernel.cc | 50 bool LogSoftmaxCPUKernel::Launch(const std::vector<kernel::AddressPtr> &inputs, const std::vector<k… in Launch() function in mindspore::kernel::LogSoftmaxCPUKernel
|
| D | log_softmax_grad_cpu_kernel.cc | 55 bool LogSoftmaxGradCPUKernel::Launch(const std::vector<kernel::AddressPtr> &inputs, in Launch() function in mindspore::kernel::LogSoftmaxGradCPUKernel
|
| D | assignadd_cpu_kernel.cc | 57 bool AssignAddCPUKernel::Launch(const std::vector<kernel::AddressPtr> &inputs, const std::vector<ke… in Launch() function in mindspore::kernel::AssignAddCPUKernel
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | assign.cc | 29 bool AssignKernel::Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> & /*… in Launch() function in mindspore::kernel::AssignKernel
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | strided_slice_gpu_kernel.h | 39 bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &, in Launch() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/nvtit/ |
| D | md_iteration_gradient_descent_kernel.h | 51 bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspace, in Launch() function
|