Searched refs:new_spatial_dim (Results 1 – 1 of 1) sorted by relevance
488 int64 new_batch_dim, new_spatial_dim; in BringSpaceNextToBatch() local500 new_spatial_dim = pushed_counter; in BringSpaceNextToBatch()517 spatial_dimension_to_split = new_spatial_dim; in BringSpaceNextToBatch()532 new_spatial_dim = pushed_counter; in BringSpaceNextToBatch()551 spatial_dimension_to_split = new_spatial_dim; in BringSpaceNextToBatch()