Home
last modified time | relevance | path

Searched refs:new_spatial_dim_size (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc170 int64 spatial_dimension, int64 new_spatial_dim_size);
571 int64 spatial_dimension, int64 new_spatial_dim_size) { in IncreaseSpatialSizeOnSpaceToBatchedShape() argument
594 ->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()