Searched refs:axis_shape_size (Results 1 – 1 of 1) sorted by relevance
25 int axis_shape_size = input_shape[axis]; in SoftmaxInt8() local36 int outter_offset = o * axis_shape_size * inner_size; in SoftmaxInt8()40 for (int i = 0; i < axis_shape_size; ++i) { in SoftmaxInt8()46 for (int i = 0; i < axis_shape_size; ++i) { in SoftmaxInt8()57 for (int i = 0; i < axis_shape_size; ++i) { in SoftmaxInt8()