Searched refs:launch_mul_ (Results 1 – 4 of 4) sorted by relevance
94 if (launch_mul_ == nullptr) { in CalculateMean()95 launch_mul_ = CreateLaunchMul(); in CalculateMean()96 MS_EXCEPTION_IF_NULL(launch_mul_); in CalculateMean()99 launch_mul_->SetInputAddr(ar_output_addr_); in CalculateMean()101 launch_mul_->LaunchOpKernel(); in CalculateMean()103 auto launch_output = launch_mul_->GetKernelOutputAddr(); in CalculateMean()
41 launch_mul_(nullptr), in Bucket()64 std::shared_ptr<LaunchKernel> launch_mul_; variable
95 if (launch_mul_ != nullptr) { in FreeAllDeviceMem()96 launch_mul_->FreeLaunchDeviceMem(); in FreeAllDeviceMem()
89 if (launch_mul_ != nullptr) { in FreeAllDeviceMem()90 launch_mul_->FreeLaunchDeviceMem(); in FreeAllDeviceMem()