Searched refs:cur_head (Results 1 – 1 of 1) sorted by relevance
92 int cur_head = perm[0]; in ReduceTransposeDimensions() local93 new_dim_position[cur_head] = 0; in ReduceTransposeDimensions()94 combined_dims[0] = shape.dim_size(cur_head); in ReduceTransposeDimensions()99 if (cur_head + 1 == perm[perm_idx]) { in ReduceTransposeDimensions()100 cur_head = perm[perm_idx]; in ReduceTransposeDimensions()101 combined_dims[dim_idx] *= shape.dim_size(cur_head); in ReduceTransposeDimensions()104 cur_head = perm[perm_idx]; in ReduceTransposeDimensions()106 new_dim_position[cur_head] = dim_idx; in ReduceTransposeDimensions()107 combined_dims[dim_idx] = shape.dim_size(cur_head); in ReduceTransposeDimensions()