|
|
|
@@ -91,8 +91,7 @@ int ArithmeticCPUKernel::ConstTensorBroadCast() { |
|
|
|
param_->in_elements_num1_ != param_->out_elements_num_) { |
|
|
|
return RET_OK; |
|
|
|
} |
|
|
|
if ((param_->in_elements_num0_ == 1 || param_->in_elements_num1_ == 1) && |
|
|
|
(arithmetic_opt_run_ != nullptr && arithmetic_opt_run_int_ != nullptr)) { |
|
|
|
if ((param_->in_elements_num0_ == 1 || param_->in_elements_num1_ == 1) && arithmetic_opt_run_ != nullptr) { |
|
|
|
/* run opt function |
|
|
|
* one of input is scalar */ |
|
|
|
return RET_OK; |
|
|
|
@@ -327,8 +326,7 @@ int ArithmeticCPUKernel::DoArithmetic(int task_id) { |
|
|
|
} |
|
|
|
int offset = stride * task_id * data_type_len_; |
|
|
|
/* run opt function, one of input is scalar */ |
|
|
|
if ((param_->in_elements_num0_ == 1 || param_->in_elements_num1_ == 1) && |
|
|
|
(arithmetic_opt_run_ != nullptr && arithmetic_opt_run_int_ != nullptr)) { |
|
|
|
if ((param_->in_elements_num0_ == 1 || param_->in_elements_num1_ == 1) && arithmetic_opt_run_ != nullptr) { |
|
|
|
if (param_->in_elements_num0_ == 1) { |
|
|
|
return Execute(input0_ptr_, static_cast<uint8_t *>(input1_ptr_) + offset, |
|
|
|
static_cast<uint8_t *>(output_ptr_) + offset, count, true); |
|
|
|
|