Home
last modified time | relevance | path

Searched refs:CreateLaunchMul (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_bucket.h38 std::shared_ptr<LaunchKernel> CreateLaunchMul() override;
Dgpu_bucket.cc159 std::shared_ptr<LaunchKernel> GPUBucket::CreateLaunchMul() { in CreateLaunchMul() function in mindspore::device::gpu::GPUBucket
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_bucket.h38 std::shared_ptr<LaunchKernel> CreateLaunchMul() override;
Dascend_bucket.cc169 std::shared_ptr<LaunchKernel> AscendBucket::CreateLaunchMul() { in CreateLaunchMul() function in mindspore::device::ascend::AscendBucket
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dbucket.h82 virtual std::shared_ptr<LaunchKernel> CreateLaunchMul() = 0;
Dbucket.cc95 launch_mul_ = CreateLaunchMul(); in CalculateMean()