Searched refs:target_dims_stack (Results 1 – 1 of 1) sorted by relevance
317 std::vector<int64> target_dims_stack(target_shape.rank()); in ReshapeSharding() local325 target_dims_stack[i] = target_shape.dimensions(target_shape.rank() - 1 - i); in ReshapeSharding()327 while (!source_dims_stack.empty() || !target_dims_stack.empty()) { in ReshapeSharding()328 if (target_dims_stack.empty()) { in ReshapeSharding()343 t_size = target_dims_stack.back(); in ReshapeSharding()344 target_dims_stack.pop_back(); in ReshapeSharding()363 target_dims_stack.push_back(t_size); in ReshapeSharding()397 target_dims_stack.push_back(t_size); in ReshapeSharding()