Searched refs:offsets_on_padded_shape (Results 1 – 1 of 1) sorted by relevance
579 std::vector<HloInstruction*> offsets_on_padded_shape(base_shape_.rank()); in ReshardAsWindowedInput() local586 offsets_on_padded_shape[i] = state_.b->AddInstruction( in ReshardAsWindowedInput()632 offsets_on_padded_shape[i] = start_on_padded_calculations[i].Calculate( in ReshardAsWindowedInput()761 shard_shape, padded_hlo, offsets_on_padded_shape, in ReshardAsWindowedInput()808 offsets_on_padded_shape[dim], pad_value, partition_ordinals[dim], in ReshardAsWindowedInput()