Searched refs:new_layout_specs (Results 1 – 1 of 1) sorted by relevance
202 std::vector<ShardingSpec> new_layout_specs(input_rank); in ExpandOp() local205 new_layout_specs[input_index].set_sharding_spec(Layout::kUnshardedDim); in ExpandOp()207 new_layout_specs[input_index] = input_layout->dim(input_index); in ExpandOp()212 Layout::GetLayout(std::move(new_layout_specs), input_layout->mesh())); in ExpandOp()