Searched refs:remapped_batch_dimension (Results 1 – 1 of 1) sorted by relevance
541 int64_t remapped_batch_dimension = spatial_dimension_to_split - 1; in HaloDuplicateWithSlice() local557 end_indices[remapped_batch_dimension] = batch_size - 1; in HaloDuplicateWithSlice()566 padding_config.mutable_dimensions(remapped_batch_dimension) in HaloDuplicateWithSlice()579 start_indices_halo[remapped_batch_dimension] = 1; in HaloDuplicateWithSlice()588 padding_config_halo.mutable_dimensions(remapped_batch_dimension) in HaloDuplicateWithSlice()