Searched refs:new_dims_cnt (Results 1 – 1 of 1) sorted by relevance
82 int new_dims_cnt = 0; in RemoveOneSizeDimensions() local87 input_shape->SetDim(new_dims_cnt, input_shape->Dims(i)); in RemoveOneSizeDimensions()88 ++new_dims_cnt; in RemoveOneSizeDimensions()90 input_shape->Resize(new_dims_cnt); in RemoveOneSizeDimensions()94 new_dims_cnt = 0; in RemoveOneSizeDimensions()99 new_params.perm[new_dims_cnt] = params->perm[i]; in RemoveOneSizeDimensions()100 output_shape->SetDim(new_dims_cnt, output_shape->Dims(i)); in RemoveOneSizeDimensions()101 ++new_dims_cnt; in RemoveOneSizeDimensions()103 output_shape->Resize(new_dims_cnt); in RemoveOneSizeDimensions()104 new_params.perm_count = new_dims_cnt; in RemoveOneSizeDimensions()[all …]