| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | sort_gpu_kernel.cc | 25 bool SortGpuKernelMod<K, V>::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::SortGpuKernelMod
33 bool SortGpuKernelMod<int32_t, half>::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::SortGpuKernelMod
109 bool SortGpuKernelMod<int32_t, float>::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::SortGpuKernelMod
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | fill_cpu_kernel.cc | 52 bool FillCpuKernelMod::LaunchKernel(const std::vector<kernel::KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::FillCpuKernelMod
97 bool FillCpuKernelMod::LaunchKernel<std::complex<float>>(const std::vector<kernel::KernelTensor *> … in LaunchKernel() function in mindspore::kernel::FillCpuKernelMod
123 bool FillCpuKernelMod::LaunchKernel<std::complex<double>>(const std::vector<kernel::KernelTensor *>… in LaunchKernel() function in mindspore::kernel::FillCpuKernelMod
|
| D | shape_cpu_kernel.cc | 44 bool ShapeCpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, const std::vector<K… in LaunchKernel() function in mindspore::kernel::ShapeCpuKernelMod
|
| D | unique_cpu_kernel.cc | 60 void UniqueCpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::UniqueCpuKernelMod
|
| D | elu_cpu_kernel.cc | 24 bool EluCpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::EluCpuKernelMod
|
| D | elu_grad_cpu_kernel.cc | 26 bool EluGradCpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::EluGradCpuKernelMod
|
| D | nextafter_cpu_kernel.cc | 37 bool NextAfterCpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, const std::vect… in LaunchKernel() function in mindspore::kernel::NextAfterCpuKernelMod
|
| D | expm1_cpu_kernel.cc | 64 void Expm1CpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::Expm1CpuKernelMod
|
| D | range_v2_cpu_kernel.cc | 34 bool RangeV2CpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, const std::vector… in LaunchKernel() function in mindspore::kernel::RangeV2CpuKernelMod
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/rpc/ |
| D | mux_send_actor.cc | 22 bool MuxSendActor::LaunchKernel(OpContext<DeviceTensor> *const context, bool is_skip_launch) { in LaunchKernel() function in mindspore::runtime::MuxSendActor
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | launch_kernel.h | 28 explicit LaunchKernel(void *stream) : stream_(stream), kernel_mod_(nullptr) {} in LaunchKernel() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | kernel_async_launch_actor.cc | 33 void KernelAsyncLaunchActor::LaunchKernel(OpContext<DeviceTensor> *const context, KernelActor *kern… in LaunchKernel() function in mindspore::runtime::KernelAsyncLaunchActor
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | identity_gpu_kernel.cc | 38 bool IdentityGpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::IdentityGpuKernelMod
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | momentum_gpu_kernel.cc | 23 void MomentumGpuKernelMod::LaunchKernel(const std::vector<kernel::KernelTensor *> &inputs, void *st… in LaunchKernel() function in mindspore::kernel::MomentumGpuKernelMod
|
| D | apply_gradient_descent_gpu_kernel.cc | 23 void ApplyGradientDescentKernelMod::LaunchKernel(const std::vector<kernel::KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::ApplyGradientDescentKernelMod
|
| D | l2_loss_gpu_kernel.cc | 47 bool L2LossGpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::L2LossGpuKernelMod
|
| D | celu_gpu_kernel.cc | 53 bool CeluGpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, const std::vector<Ke… in LaunchKernel() function in mindspore::kernel::CeluGpuKernelMod
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/bisheng/ |
| D | add_bisheng_kernel.cc | 44 bool AddBishengKernel::LaunchKernel(const std::vector<kernel::KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::AddBishengKernel
|
| D | wkv_bisheng_kernel.cc | 43 bool WKVBishengKernel::LaunchKernel(const std::vector<kernel::KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::WKVBishengKernel
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/sequence/ |
| D | sequence_len_cpu_kernel.cc | 47 bool SequenceLenCpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &, const std::vector<K… in LaunchKernel() function in mindspore::kernel::SequenceLenCpuKernelMod
|
| D | sequence_min_cpu_kernel.cc | 46 bool SequenceMinCpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::SequenceMinCpuKernelMod
|
| D | sequence_zeros_like_cpu_kernel.cc | 44 bool SequenceZerosLikeCpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &, in LaunchKernel() function in mindspore::kernel::SequenceZerosLikeCpuKernelMod
|
| D | sequence_max_cpu_kernel.cc | 46 bool SequenceMaxCpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::SequenceMaxCpuKernelMod
|
| D | shape_mul_cpu_kernel.cc | 49 bool ShapeMulCpuKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, const std::vecto… in LaunchKernel() function in mindspore::kernel::ShapeMulCpuKernelMod
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/other/ |
| D | dynamic_reshape_gpu_kernel.cc | 50 bool DynamicReshapeKernelMod::LaunchKernel(const std::vector<KernelTensor *> &inputs, in LaunchKernel() function in mindspore::kernel::DynamicReshapeKernelMod
|