Searched refs:new_spatial_dim_size (Results 1 – 1 of 1) sorted by relevance
170 int64 spatial_dimension, int64 new_spatial_dim_size);571 int64 spatial_dimension, int64 new_spatial_dim_size) { in IncreaseSpatialSizeOnSpaceToBatchedShape() argument594 ->set_edge_padding_high(new_spatial_dim_size * new_batch_size / in IncreaseSpatialSizeOnSpaceToBatchedShape()610 reshape_back_dims[spatial_dimension] = new_spatial_dim_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()