Searched refs:skip_dims_cnt (Results 1 – 1 of 1) sorted by relevance
126 int skip_dims_cnt = 0; in Flatten() local131 ++skip_dims_cnt; in Flatten()138 const int new_dims_cnt = params.perm_count - skip_dims_cnt; in Flatten()143 for (int i = skip_dims_cnt; i < params.perm_count; ++i) { in Flatten()144 non_flatten_input_shape->SetDim(i - skip_dims_cnt, input_shape.Dims(i)); in Flatten()145 non_flatten_output_shape->SetDim(i - skip_dims_cnt, output_shape.Dims(i)); in Flatten()146 non_flatten_params->perm[i - skip_dims_cnt] = params.perm[i]; in Flatten()