Searched refs:outter_size (Results 1 – 9 of 9) sorted by relevance
29 int inner_size = 1, outter_size = 1; in SoftmaxGrad() local31 outter_size *= input_shape[i]; in SoftmaxGrad()36 NNACL_CHECK_ZERO_RETURN(outter_size); in SoftmaxGrad()39 dim /= outter_size; in SoftmaxGrad()44 for (int i = 0; i < outter_size; i++) { in SoftmaxGrad()
51 int outter_size = 1; in LogSoftmax() local54 outter_size *= input_shape[i]; in LogSoftmax()59 for (int i = 0; i < outter_size; i++) { in LogSoftmax()77 for (int i = 0; i < outter_size; i++) { in LogSoftmax()
110 int outter_size = 1; in Softmax() local113 outter_size *= input_shape[i]; in Softmax()118 for (int i = 0; i < outter_size; i++) { in Softmax()136 for (int i = 0; i < outter_size; i++) { in Softmax()
66 int outter_size = 1; in DoCode() local68 outter_size *= softmax_param_->input_shape_[i]; in DoCode()91 int stride = UP_DIV(outter_size, thread_num_); in DoCode()92 int count = MSMIN(stride, outter_size - stride * kDefaultTaskId); in DoCode()
54 int outter_size = 1; in LogSoftmaxFp16() local57 outter_size *= input_shape[i]; in LogSoftmaxFp16()62 for (int i = 0; i < outter_size; i++) { in LogSoftmaxFp16()80 for (int i = 0; i < outter_size; i++) { in LogSoftmaxFp16()
100 int outter_size = 1; in SoftmaxFp16() local103 outter_size *= input_shape[i]; in SoftmaxFp16()108 for (int i = 0; i < outter_size; i++) { in SoftmaxFp16()126 for (int i = 0; i < outter_size; i++) { in SoftmaxFp16()
91 int outter_size = 1, inner_size = 1; in DoSoftmax() local93 outter_size *= softmax_param_->input_shape_[i]; in DoSoftmax()99 int stride = UP_DIV(outter_size, thread_count_); in DoSoftmax()104 int count = MSMIN(stride, outter_size - stride * task_id); in DoSoftmax()
110 int outter_size = 1; in Softmax() local113 outter_size *= input_shape[i]; in Softmax()118 for (int i = 0; i < outter_size; i++) { in Softmax()135 for (int i = 0; i < outter_size; i++) { in Softmax()
133 int outter_size = 1; in Run() local142 outter_size *= input_shape[i]; in Run()148 outter_size_ = outter_size; in Run()