Searched refs:spatial_dimension_to_split (Results 1 – 1 of 1) sorted by relevance
540 int64_t spatial_dimension_to_split = activations_batch_dim + 2 * (i + 1); in HaloDuplicateWithSlice() local541 int64_t remapped_batch_dimension = spatial_dimension_to_split - 1; in HaloDuplicateWithSlice()555 start_indices[spatial_dimension_to_split] = in HaloDuplicateWithSlice()558 end_indices[spatial_dimension_to_split] = spatial_split_size; in HaloDuplicateWithSlice()580 end_indices_halo[spatial_dimension_to_split] = halo_size - low_padding; in HaloDuplicateWithSlice()600 end_indices_activations_cut[spatial_dimension_to_split] = in HaloDuplicateWithSlice()603 start_indices_activations_cut[spatial_dimension_to_split] = in HaloDuplicateWithSlice()605 end_indices_activations_cut[spatial_dimension_to_split] = in HaloDuplicateWithSlice()617 spatial_dimension_to_split)); in HaloDuplicateWithSlice()623 spatial_dimension_to_split)); in HaloDuplicateWithSlice()[all …]