Searched refs:new_spatial_dim_size (Results 1 – 1 of 1) sorted by relevance
215 int64_t new_spatial_dim_size, bool increase_spatial_size = false);798 int64_t new_spatial_dim_size, bool increase_spatial_size) { in ChangeSpatialSizeOnSpaceToBatchedShape() argument843 ->set_edge_padding_high(new_spatial_dim_size * in ChangeSpatialSizeOnSpaceToBatchedShape()863 new_spatial_dim_size * ctrl_.number_of_splits; in ChangeSpatialSizeOnSpaceToBatchedShape()875 batch_dimension, new_spatial_dim_size, in ChangeSpatialSizeOnSpaceToBatchedShape()