Searched refs:compute_units_ (Results 1 – 7 of 7) sorted by relevance
35 DeConvComputeUnit &unit = deconv_param_->compute_units_[i]; in FreeResizeBuf()83 DeConvComputeUnit &unit = deconv_param_->compute_units_[i]; in FreeDeconvParam()102 if (deconv_param_->compute_units_ != nullptr) { in FreeDeconvParam()103 free(deconv_param_->compute_units_); in FreeDeconvParam()104 deconv_param_->compute_units_ = nullptr; in FreeDeconvParam()152 DeConvComputeUnit &unit = deconv_param_->compute_units_[i]; in InitParameter()275 deconv_param_->compute_units_ = reinterpret_cast<DeConvComputeUnit *>(malloc(size)); in InitComputeParam()276 if (deconv_param_->compute_units_ == nullptr) { in InitComputeParam()322 deconv_param_->compute_units_[cur_count] = unit; in InitComputeParam()340 DeConvComputeUnit *unit = &deconv_param_->compute_units_[i]; in InitDataParam()
37 DeConvComputeUnit &unit = deconv_param_->compute_units_[i]; in FreeResizeBuf()76 DeConvComputeUnit &unit = deconv_param_->compute_units_[i]; in FreeDeconvParam()95 if (deconv_param_->compute_units_ != nullptr) { in FreeDeconvParam()96 free(deconv_param_->compute_units_); in FreeDeconvParam()97 deconv_param_->compute_units_ = nullptr; in FreeDeconvParam()134 DeConvComputeUnit &unit = deconv_param_->compute_units_[i]; in InitParameter()228 deconv_param_->compute_units_ = reinterpret_cast<DeConvComputeUnit *>(malloc(size)); in InitComputeParam()229 if (deconv_param_->compute_units_ == nullptr) { in InitComputeParam()281 deconv_param_->compute_units_[cur_count] = unit; in InitComputeParam()300 DeConvComputeUnit *unit = &deconv_param_->compute_units_[i]; in InitDataParam()
117 DeConvComputeUnit *compute_units_; member
124 ret = device_->getInfo(CL_DEVICE_MAX_COMPUTE_UNITS, &compute_units_); in InitGPUDevice()169 MS_LOG(INFO) << "Compute Unit: " << compute_units_; in InitGPUDevice()342 uint32_t OpenCLRuntime::DeviceComputeUnits() const { return compute_units_; } in DeviceComputeUnits()
182 uint32_t compute_units_{0};
555 DeConvComputeUnit *unit = &deconv_param->compute_units_[i]; in DeconvWg()
379 DeConvComputeUnit *unit = &deconv_param->compute_units_[i]; in DeconvWgFp16()