Searched refs:reshard_output (Results 1 – 1 of 1) sorted by relevance
2126 auto reshard_output = tmp_output.ReshardAsWindowedInput( in HandleReshape() local2130 if (!reshard_output.has_value()) { in HandleReshape()2133 TF_RET_CHECK(!reshard_output->dynamic_slice_index_on_output.has_value()); in HandleReshape()2135 reshard_output->sharded_input->shape().dimensions(output_sharded_dim), in HandleReshape()2137 SetPartitionedHlo(hlo, [&] { return reshard_output->sharded_input; }); in HandleReshape()