Searched refs:spatial_split_size (Results 1 – 1 of 1) sorted by relevance
119 int64 high_padding, int64 low_padding, int64 spatial_split_size,127 int64 spatial_split_size, int64 num_splits);367 const int64 spatial_split_size = in HaloDuplicateWithSlice() local372 CHECK_LE(std::abs(halo_size - low_padding), spatial_split_size); in HaloDuplicateWithSlice()375 << " spatial_split_size " << spatial_split_size << " low_padding " in HaloDuplicateWithSlice()392 spatial_split_size - low_padding; in HaloDuplicateWithSlice()394 end_indices[spatial_dimension_to_split] = spatial_split_size; in HaloDuplicateWithSlice()436 spatial_split_size - low_padding; in HaloDuplicateWithSlice()441 spatial_split_size; in HaloDuplicateWithSlice()1915 int64 spatial_split_size = in PropagateOnConv() local[all …]