Searched refs:spatial_split_size (Results 1 – 1 of 1) sorted by relevance
130 int64_t spatial_split_size, int64_t num_splits,138 int64_t activations_batch_dim, int64_t spatial_split_size,156 int64_t spatial_split_size, int64_t num_splits);521 const int64_t spatial_split_size = in HaloDuplicateWithSlice() local534 << " spatial_split_size " << spatial_split_size << " low_padding " in HaloDuplicateWithSlice()537 CHECK_LE(std::abs(halo_size - low_padding), spatial_split_size); in HaloDuplicateWithSlice()556 spatial_split_size - low_padding; in HaloDuplicateWithSlice()558 end_indices[spatial_dimension_to_split] = spatial_split_size; in HaloDuplicateWithSlice()601 spatial_split_size - low_padding; in HaloDuplicateWithSlice()606 spatial_split_size; in HaloDuplicateWithSlice()[all …]