Home
last modified time | relevance | path

Searched defs:GenTask (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Dassign.cc51 std::vector<TaskInfoPtr> AssignKernel::GenTask(const std::vector<AddressPtr> &inputs, const std::ve… in GenTask() function in mindspore::kernel::AssignKernel
Dlabel_set.cc51 std::vector<TaskInfoPtr> LabelSetKernel::GenTask(const std::vector<AddressPtr> &, const std::vector… in GenTask() function in mindspore::kernel::LabelSetKernel
Dsend.cc54 std::vector<TaskInfoPtr> SendKernel::GenTask(const std::vector<AddressPtr> &, const std::vector<Add… in GenTask() function in mindspore::kernel::SendKernel
Drecv.cc55 std::vector<TaskInfoPtr> RecvKernel::GenTask(const std::vector<AddressPtr> &, const std::vector<Add… in GenTask() function in mindspore::kernel::RecvKernel
Dlabel_goto.cc51 std::vector<TaskInfoPtr> LabelGotoKernel::GenTask(const std::vector<AddressPtr> &, const std::vecto… in GenTask() function in mindspore::kernel::LabelGotoKernel
Dprofiling_kernel_mod.cc57 std::vector<TaskInfoPtr> ProfilingKernelMod::GenTask(const std::vector<AddressPtr> &inputs, in GenTask() function in mindspore::kernel::ProfilingKernelMod
Dstream_active.cc70 std::vector<TaskInfoPtr> StreamActiveKernel::GenTask(const std::vector<AddressPtr> &, const std::ve… in GenTask() function in mindspore::kernel::StreamActiveKernel
Dstream_switch.cc87 std::vector<TaskInfoPtr> StreamSwitchKernel::GenTask(const std::vector<AddressPtr> &inputs, in GenTask() function in mindspore::kernel::StreamSwitchKernel
Dlabel_switch.cc59 std::vector<TaskInfoPtr> LabelSwitchKernel::GenTask(const std::vector<AddressPtr> &inputs, in GenTask() function in mindspore::kernel::LabelSwitchKernel
Dmemcpy_async.cc106 std::vector<TaskInfoPtr> MemCpyAsyncKernel::GenTask(const std::vector<AddressPtr> &inputs, in GenTask() function in mindspore::kernel::MemCpyAsyncKernel
Dtensor_copy_slices.cc134 std::vector<TaskInfoPtr> TensorCopySlices::GenTask(const std::vector<AddressPtr> &inputs, in GenTask() function in mindspore::kernel::TensorCopySlices
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhcom_all_to_all.cc63 std::vector<TaskInfoPtr> HcomAllToAllKernel::GenTask(const std::vector<AddressPtr> &inputs, in GenTask() function in mindspore::kernel::HcomAllToAllKernel
Dhccl_kernel.cc209 std::vector<TaskInfoPtr> HcclKernel::GenTask(const std::vector<AddressPtr> &inputs, in GenTask() function in mindspore::kernel::HcclKernel
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_mod.cc79 std::vector<TaskInfoPtr> HostKernelMod::GenTask(const std::vector<AddressPtr> &, const std::vector<… in GenTask() function in mindspore::kernel::HostKernelMod
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/
Dakg_ascend_kernel_mod.cc95 std::vector<TaskInfoPtr> AkgKernelMod::GenTask(const std::vector<AddressPtr> &inputs, in GenTask() function in mindspore::kernel::AkgKernelMod
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_mod.cc165 std::vector<TaskInfoPtr> AicpuOpKernelMod::GenTask(const std::vector<AddressPtr> &inputs, in GenTask() function in mindspore::kernel::AicpuOpKernelMod
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.cc73 std::vector<TaskInfoPtr> TbeKernelMod::GenTask(const std::vector<AddressPtr> &inputs, in GenTask() function in mindspore::kernel::TbeKernelMod
/third_party/mindspore/tests/ut/cpp/stub/ge/
Dge_task_launch_stub.cc34 bool HcclAdapter::GenTask(const AnfNodePtr &, HcclDataType, std::vector<HcclTaskInfo> *) const { re… in GenTask() function in mindspore::hccl::HcclAdapter
/third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/
Dhccl_adapter.cc203 bool HcclAdapter::GenTask(const AnfNodePtr &node, HcclDataType datatype, in GenTask() function in mindspore::hccl::HcclAdapter
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc450 bool AscendKernelRuntime::GenTask(const session::KernelGraph &graph) { in GenTask() function in mindspore::device::ascend::AscendKernelRuntime