Home
last modified time | relevance | path

Searched defs:LaunchKernel (Results 1 – 25 of 49) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_kernel.h27 explicit LaunchKernel(void *stream) : stream_(stream), kernel_mod_(nullptr) {} in LaunchKernel() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dexpm1_cpu_kernel.cc52 void Expm1CPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::Expm1CPUKernel
Drange_cpu_kernel.cc47 bool RangeCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, const std::vector<AddressP… in LaunchKernel() function in mindspore::kernel::RangeCPUKernel
Dunique_cpu_kernel.cc74 void UniqueCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, const std::vector<Address… in LaunchKernel() function in mindspore::kernel::UniqueCPUKernel
Ddropout_cpu_kernel.cc63 void DropoutCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::DropoutCPUKernel
Dmap_uniform_cpu_kernel.cc52 void MapUniformCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::MapUniformCPUKernel
Dsigmoid_cross_entropy_with_logits_cpu_kernel.cc51 void SigmoidCrossEntropyWithLogitsCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::SigmoidCrossEntropyWithLogitsCPUKernel
Dsigmoid_cross_entropy_with_logits_grad_cpu_kernel.cc53 void SigmoidCrossEntropyWithLogitsGradCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::SigmoidCrossEntropyWithLogitsGradCPUKernel
Delu_grad_cpu_kernel.cc54 void EluGradCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::EluGradCPUKernel
Dunpack_cpu_kernel.cc74 void UnpackCPUKernel<T>::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::UnpackCPUKernel
Dupdate_cache_cpu_kernel.cc61 void UpdateCacheCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::UpdateCacheCPUKernel
Dresize_nearest_neighbor_grad_cpu_kernel.cc80 void ResizeNearestNeighborGradCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::ResizeNearestNeighborGradCPUKernel
Dresize_nearest_neighbor_cpu_kernel.cc79 void ResizeNearestNeighborCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::ResizeNearestNeighborCPUKernel
Dsub_and_filter_cpu_kernel.cc51 void SubAndFilterCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::SubAndFilterCPUKernel
Ddynamic_assign_cpu_kernel.cc59 void DynamicAssignCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::DynamicAssignCPUKernel
Dpad_and_shift_cpu_kernel.cc61 void PadAndShiftCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::PadAndShiftCPUKernel
Dtile_cpu_kernel.cc123 void TileCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, const std::vector<AddressPt… in LaunchKernel() function in mindspore::kernel::TileCPUKernel
Dpad_cpu_kernel.cc82 void PadCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr… in LaunchKernel() function in mindspore::kernel::PadCPUKernel
Dbinary_cross_entropy_grad_kernel.cc27 void BinaryCrossEntropyGradCpuKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::BinaryCrossEntropyGradCpuKernel
Dtranspose_cpu_kernel.cc86 void TransposeCPUFwdKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::TransposeCPUFwdKernel
Dtopk_cpu_kernel.cc30 void TopKCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, const std::vector<AddressPt… in LaunchKernel() function in mindspore::kernel::TopKCPUKernel
Dapply_adagrad_cpu_kernel.cc66 void ApplyAdagradCPUKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::ApplyAdagradCPUKernel
Dsplit_cpu_kernel.cc91 void SplitCPUKernel<T>::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::SplitCPUKernel
Dbinary_cross_entropy_cpu_kernel.cc49 void BinaryCrossEntropyCpuKernel::LaunchKernel(const std::vector<AddressPtr> &inputs, in LaunchKernel() function in mindspore::kernel::BinaryCrossEntropyCpuKernel
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_device_address.h34 class LaunchKernel; variable

12