Searched refs:axis_offset (Results 1 – 10 of 10) sorted by relevance
45 int axis_offset = inner_offset + j; in SoftMaxP1Simple() local46 max_data = max_data > input_ptr[axis_offset] ? max_data : input_ptr[axis_offset]; in SoftMaxP1Simple()51 int axis_offset = inner_offset + j; in SoftMaxP1Simple() local52 _sum_data += output_ptr[axis_offset]; in SoftMaxP1Simple()71 int axis_offset = inner_offset + j * inner_size; in SoftMaxP1() local72 max_data = max_data > input_ptr[axis_offset] ? max_data : input_ptr[axis_offset]; in SoftMaxP1()75 int axis_offset = inner_offset + j * inner_size; in SoftMaxP1() local76 output_ptr[axis_offset] = exp(input_ptr[axis_offset] - max_data); in SoftMaxP1()80 int axis_offset = inner_offset + j * inner_size; in SoftMaxP1() local81 _sum_data += output_ptr[axis_offset]; in SoftMaxP1()[all …]
67 int axis_offset = inner_offset + j * inner_size; in LogSoftmax() local68 max_data = max_data > input_ptr[axis_offset] ? max_data : input_ptr[axis_offset]; in LogSoftmax()71 int axis_offset = inner_offset + j * inner_size; in LogSoftmax() local72 output_ptr[axis_offset] = input_ptr[axis_offset] - max_data; in LogSoftmax()73 sum_data[k + sum_outter_offset] += expf(output_ptr[axis_offset]); in LogSoftmax()81 int axis_offset = outter_offset + j * inner_size; in LogSoftmax() local83 int inner_offset = axis_offset + k; in LogSoftmax()
126 int axis_offset = inner_offset + j * inner_size; in Softmax() local127 max_data = max_data > input_ptr[axis_offset] ? max_data : input_ptr[axis_offset]; in Softmax()130 int axis_offset = inner_offset + j * inner_size; in Softmax() local131 output_ptr[axis_offset] = expf(input_ptr[axis_offset] - max_data); in Softmax()132 sum_data[k + sum_outter_offset] += output_ptr[axis_offset]; in Softmax()140 int axis_offset = outter_offset + j * inner_size; in Softmax() local142 int inner_offset = axis_offset + k; in Softmax()
26 int axis_offset = out_offset + i * inner_size; in ScaleInner() local32 int in_offset = axis_offset + in_index; in ScaleInner()42 int in_offset = axis_offset + in_index; in ScaleInner()49 int in_offset = axis_offset + in_index; in ScaleInner()116 int axis_offset = out_offset + i * inner_size; in ScaleInnerRelu() local122 int in_offset = axis_offset + in_index; in ScaleInnerRelu()133 int in_offset = axis_offset + in_index; in ScaleInnerRelu()141 int in_offset = axis_offset + in_index; in ScaleInnerRelu()220 int axis_offset = out_offset + i * inner_size; in ScaleInnerRelu6() local226 int in_offset = axis_offset + in_index; in ScaleInnerRelu6()[all …]
41 int axis_offset = outter_offset + c + i * inner_size; in SoftmaxInt8() local42 max_row = MSMAX(max_row, input_ptr[axis_offset]); in SoftmaxInt8()47 int axis_offset = outter_offset + c + i * inner_size; in SoftmaxInt8() local48 const int32_t input_val = input_ptr[axis_offset] - max_row; in SoftmaxInt8()52 exp_data[axis_offset] = exp_val; in SoftmaxInt8()58 int axis_offset = outter_offset + i * inner_size; in SoftmaxInt8() local63 …SaturatingRoundingDoublingHighMul(shifted_scale, exp_data[axis_offset + c]), num_bits_over_unit + … in SoftmaxInt8()66 output_ptr[axis_offset + c] = in SoftmaxInt8()
70 int axis_offset = inner_offset + j * inner_size; in LogSoftmaxFp16() local71 max_data = max_data > input_ptr[axis_offset] ? max_data : input_ptr[axis_offset]; in LogSoftmaxFp16()74 int axis_offset = inner_offset + j * inner_size; in LogSoftmaxFp16() local75 output_ptr[axis_offset] = input_ptr[axis_offset] - max_data; in LogSoftmaxFp16()76 sum_data[k + sum_outter_offset] += exp(output_ptr[axis_offset]); in LogSoftmaxFp16()84 int axis_offset = outter_offset + j * inner_size; in LogSoftmaxFp16() local86 int inner_offset = axis_offset + k; in LogSoftmaxFp16()
116 int axis_offset = inner_offset + j * inner_size; in SoftmaxFp16() local117 max_data = max_data > input_ptr[axis_offset] ? max_data : input_ptr[axis_offset]; in SoftmaxFp16()120 int axis_offset = inner_offset + j * inner_size; in SoftmaxFp16() local121 output_ptr[axis_offset] = expf(input_ptr[axis_offset] - max_data); in SoftmaxFp16()122 sum_data[k + sum_outter_offset] += output_ptr[axis_offset]; in SoftmaxFp16()130 int axis_offset = outter_offset + j * inner_size; in SoftmaxFp16() local132 int inner_offset = axis_offset + k; in SoftmaxFp16()
24 int axis_offset = out_offset + i * inner_size; in Fp16ScaleInner() local28 int in_offset = axis_offset + in_index; in Fp16ScaleInner()38 int in_offset = axis_offset + in_index; in Fp16ScaleInner()90 int axis_offset = out_offset + i * inner_size; in Fp16ScaleInnerRelu() local94 int in_offset = axis_offset + in_index; in Fp16ScaleInnerRelu()104 int in_offset = axis_offset + in_index; in Fp16ScaleInnerRelu()163 int axis_offset = out_offset + i * inner_size; in Fp16ScaleInnerRelu6() local167 int in_offset = axis_offset + in_index; in Fp16ScaleInnerRelu6()177 int in_offset = axis_offset + in_index; in Fp16ScaleInnerRelu6()
125 int axis_offset = inner_offset + j * inner_size; in Softmax() local126 max_data = max_data > input_ptr[axis_offset] ? max_data : input_ptr[axis_offset]; in Softmax()129 int axis_offset = inner_offset + j * inner_size; in Softmax() local130 output_ptr[axis_offset] = exp(input_ptr[axis_offset] - max_data); in Softmax()131 sum_data[k + sum_outter_offset] += output_ptr[axis_offset]; in Softmax()139 int axis_offset = outter_offset + j * inner_size; in Softmax() local141 int inner_offset = axis_offset + k; in Softmax()
30 int axis_offset = 0; in Concat() local43 axis_offset += inputs_output_shape[i][axis]; in Concat()49 … uint8_t *dst = dst_base + j * output_stride + axis_offset * after_axis_size + task_id * offset; in Concat()52 axis_offset += inputs_output_shape[i][axis]; in Concat()