Searched refs:explicit_left_padding (Results 1 – 1 of 1) sorted by relevance
581 std::vector<int64> explicit_left_padding(base_shape_.rank()); in ReshardAsWindowedInput() local615 explicit_left_padding[i] = wd.padding_low() / wd.base_dilation(); in ReshardAsWindowedInput()750 padding_config_dim->set_edge_padding_low(explicit_left_padding[i]); in ReshardAsWindowedInput()752 explicit_left_padding[i] - in ReshardAsWindowedInput()794 input_shard_size, explicit_left_padding[dim], 1); in ReshardAsWindowedInput()800 input_shard_size, input_shard_size + explicit_left_padding[dim], 1); in ReshardAsWindowedInput()806 right_halo_size_functions[dim], explicit_left_padding[dim], in ReshardAsWindowedInput()