Home
last modified time | relevance | path

Searched refs:launch_mul (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_bucket.cc163 auto launch_mul = std::make_shared<GPULaunchMul>(stream_, tensor_type_list_[0], total_size_); in CreateLaunchMul() local
164 MS_EXCEPTION_IF_NULL(launch_mul); in CreateLaunchMul()
165 return launch_mul; in CreateLaunchMul()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_bucket.cc173 auto launch_mul = std::make_shared<AscendLaunchMul>(stream_, tensor_type_list_[0], total_size_); in CreateLaunchMul() local
174 MS_EXCEPTION_IF_NULL(launch_mul); in CreateLaunchMul()
175 return launch_mul; in CreateLaunchMul()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
DCMakeLists.txt4 "bucket.cc" "launch_kernel.cc" "launch_mul.cc"