Searched refs:out_plane (Results 1 – 8 of 8) sorted by relevance
31 int out_plane = output_w * output_h; in AvgPooling() local32 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in AvgPooling()50 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in AvgPooling()146 int out_plane = output_w * output_h; in MaxPooling() local147 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in MaxPooling()165 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in MaxPooling()
34 int out_plane = output_w * output_h; in AvgPoolingInt8() local46 for (int i = 0; i < out_plane; i++) { in AvgPoolingInt8()89 int out_plane = output_w * pooling_param->output_h_; in AvgPoolingOptInt8() local90 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in AvgPoolingOptInt8()103 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in AvgPoolingOptInt8()290 int out_plane = output_w * output_h; in MaxPoolingInt8() local301 for (int i = 0; i < out_plane; i++) { in MaxPoolingInt8()333 int out_plane = output_w * pooling_param->output_h_; in MaxPoolingWithQuantInt8() local334 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in MaxPoolingWithQuantInt8()350 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in MaxPoolingWithQuantInt8()[all …]
28 int out_plane = out_h * out_w; in ResizeBilinearInt8() local31 int8_t *out_b_ptr = output_ptr + n * out_plane * channel; in ResizeBilinearInt8()77 int out_plane = out_h * out_w; in ResizeBilinearWithFloatScaleInt8() local80 int8_t *out_b_ptr = output_ptr + n * out_plane * channel; in ResizeBilinearWithFloatScaleInt8()
47 int out_plane = out_tensor->Height() * out_tensor->Width() * out_tensor->Batch(); in PostConcat() local48 if (out_plane < 0) { in PostConcat()58 for (int i = 0; i < out_plane; ++i) { in PostConcat()
30 int out_plane = output_w * output_h; in AvgPoolingFp16() local31 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in AvgPoolingFp16()44 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in AvgPoolingFp16()267 int out_plane = output_w * output_h; in MaxPoolingFp16() local268 int out_tile_count = UP_DIV(out_plane, TILE_NUM); in MaxPoolingFp16()278 …int real_cal_num = (out_plane - cal_start_index) > TILE_NUM ? TILE_NUM : (out_plane - cal_start_in… in MaxPoolingFp16()
40 int out_plane = conv_param_->output_h_ * conv_param_->output_w_ * conv_param_->output_batch_; in PostConcat() local47 for (int i = 0; i < out_plane; ++i) { in PostConcat()
69 int out_plane = out_tensor->Height() * out_tensor->Width() * out_tensor->Batch(); in PostConcat() local77 for (int i = 0; i < out_plane; ++i) { in PostConcat()
126 int out_plane = out_tensor->Height() * out_tensor->Width(); in AdjustNumberOfThread() local127 thread_count_ = MSMIN(op_parameter_->thread_num_, UP_DIV(out_plane, row_tile_)); in AdjustNumberOfThread()