Searched refs:outter_offset (Results 1 – 8 of 8) sorted by relevance
36 int outter_offset = o * axis_shape_size * inner_size; in SoftmaxInt8() local41 int axis_offset = outter_offset + c + i * inner_size; in SoftmaxInt8()47 int axis_offset = outter_offset + c + i * inner_size; in SoftmaxInt8()58 int axis_offset = outter_offset + i * inner_size; in SoftmaxInt8()
60 int outter_offset = i * input_shape[axis] * inner_size; in LogSoftmax() local63 int inner_offset = outter_offset + k; in LogSoftmax()78 int outter_offset = i * input_shape[axis] * inner_size; in LogSoftmax() local81 int axis_offset = outter_offset + j * inner_size; in LogSoftmax()
119 int outter_offset = i * input_shape[axis] * inner_size; in Softmax() local122 int inner_offset = outter_offset + k; in Softmax()137 int outter_offset = i * input_shape[axis] * inner_size; in Softmax() local140 int axis_offset = outter_offset + j * inner_size; in Softmax()
63 int outter_offset = i * input_shape[axis] * inner_size; in LogSoftmaxFp16() local66 int inner_offset = outter_offset + k; in LogSoftmaxFp16()81 int outter_offset = i * input_shape[axis] * inner_size; in LogSoftmaxFp16() local84 int axis_offset = outter_offset + j * inner_size; in LogSoftmaxFp16()
109 int outter_offset = i * input_shape[axis] * inner_size; in SoftmaxFp16() local112 int inner_offset = outter_offset + k; in SoftmaxFp16()127 int outter_offset = i * input_shape[axis] * inner_size; in SoftmaxFp16() local130 int axis_offset = outter_offset + j * inner_size; in SoftmaxFp16()
65 int outter_offset = i * length * inner_size; in SoftMaxP1() local68 int inner_offset = outter_offset + k; in SoftMaxP1()91 int outter_offset = i * length * inner_size; in SoftMaxP2() local94 int axis_offset = outter_offset + j * inner_size; in SoftMaxP2()
45 int outter_offset = i * dim; in SoftmaxGrad() local48 int inner_offset = outter_offset + k; in SoftmaxGrad()57 *(output_ptr + outter_offset + k * N + j) += a * sum_data[j]; in SoftmaxGrad()
119 int outter_offset = i * input_shape[axis] * inner_size; in Softmax() local122 int inner_offset = outter_offset + k; in Softmax()136 int outter_offset = i * input_shape[axis] * inner_size; in Softmax() local139 int axis_offset = outter_offset + j * inner_size; in Softmax()