Searched defs:axis_offset (Results 1 – 13 of 13) sorted by relevance
46 int axis_offset = inner_offset + j; in SoftMaxP1Simple() local52 int axis_offset = inner_offset + j; in SoftMaxP1Simple() local72 int axis_offset = inner_offset + j * inner_size; in SoftMaxP1() local76 int axis_offset = inner_offset + j * inner_size; in SoftMaxP1() local81 int axis_offset = inner_offset + j * inner_size; in SoftMaxP1() local95 int axis_offset = outter_offset + j * inner_size; in SoftMaxP2() local
37 int axis_offset = outter_offset + c + i * inner_size; in SoftmaxInt8() local43 int axis_offset = outter_offset + c + i * inner_size; in SoftmaxInt8() local54 int axis_offset = outter_offset + i * inner_size; in SoftmaxInt8() local
64 int axis_offset = inner_offset + j * inner_size; in LogSoftmax() local68 int axis_offset = inner_offset + j * inner_size; in LogSoftmax() local78 int axis_offset = outter_offset + j * inner_size; in LogSoftmax() local
104 int axis_offset = inner_offset + j * inner_size; in Softmax() local108 int axis_offset = inner_offset + j * inner_size; in Softmax() local118 int axis_offset = outter_offset + j * inner_size; in Softmax() local
25 int axis_offset = out_offset + i * inner_size; in ScaleInner() local113 int axis_offset = out_offset + i * inner_size; in ScaleInnerRelu() local215 int axis_offset = out_offset + i * inner_size; in ScaleInnerRelu6() local
67 int axis_offset = inner_offset + j * inner_size; in LogSoftmaxFp16() local71 int axis_offset = inner_offset + j * inner_size; in LogSoftmaxFp16() local81 int axis_offset = outter_offset + j * inner_size; in LogSoftmaxFp16() local
113 int axis_offset = inner_offset + j * inner_size; in SoftmaxFp16() local117 int axis_offset = inner_offset + j * inner_size; in SoftmaxFp16() local127 int axis_offset = outter_offset + j * inner_size; in SoftmaxFp16() local
24 int axis_offset = out_offset + i * inner_size; in Fp16ScaleInner() local90 int axis_offset = out_offset + i * inner_size; in Fp16ScaleInnerRelu() local163 int axis_offset = out_offset + i * inner_size; in Fp16ScaleInnerRelu6() local
54 int axis_offset = inner_offset + j * inner_size; in SoftmaxFunc() local58 int axis_offset = inner_offset + j * inner_size; in SoftmaxFunc() local65 int axis_offset = outter_offset + j * inner_size; in SoftmaxFunc() local
30 int axis_offset = 0; in Concat() local
273 size_t axis_offset = 0; in Process() local
103 uint axis_offset = outer_idx * axis_inner_size + inner_idx + axis_idx * inner_size; in LargeBlockScanKernel() local