Home
last modified time | relevance | path

Searched refs:LaunchOpKernel (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_kernel.h37 void LaunchOpKernel() override = 0;
Dascend_launch_mul.h39 void LaunchOpKernel() override;
Dascend_launch_mul.cc40 void AscendLaunchMul::LaunchOpKernel() { in LaunchOpKernel() function in mindspore::device::ascend::AscendLaunchMul
Dascend_launch_atomic_clean.h42 void LaunchOpKernel() override;
Dascend_launch_transdata.h48 void LaunchOpKernel() override;
Dascend_launch_atomic_clean.cc40 void AscendLaunchAtomicClean::LaunchOpKernel() { in LaunchOpKernel() function in mindspore::device::ascend::AscendLaunchAtomicClean
Dascend_launch_transdata.cc41 void AscendLaunchTransData::LaunchOpKernel() { in LaunchOpKernel() function in mindspore::device::ascend::AscendLaunchTransData
Dascend_bucket.cc166 launch_atomic_clean_->LaunchOpKernel(); in CleanAllReduceInputAddr()
Dascend_device_address.cc282 launch_transdata_->LaunchOpKernel(); in SyncDeviceToHostAndConvertFormatBasedOnTransData()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_launch_kernel.h37 void LaunchOpKernel() override = 0;
Dgpu_launch_mul.h38 void LaunchOpKernel() override;
Dgpu_launch_mul.cc39 void GPULaunchMul::LaunchOpKernel() { in LaunchOpKernel() function in mindspore::device::gpu::GPULaunchMul
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_kernel.h41 virtual void LaunchOpKernel() = 0;
Dbucket.cc101 launch_mul_->LaunchOpKernel(); in CalculateMean()