Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc806 const int64_t reshaped_space_size = spatial_dim_size * ctrl_.number_of_splits; in ChangeSpatialSizeOnSpaceToBatchedShape() local
825 batch_space_collapse_reshape_dims[spatial_dimension] = reshaped_space_size; in ChangeSpatialSizeOnSpaceToBatchedShape()
845 reshaped_space_size); in ChangeSpatialSizeOnSpaceToBatchedShape()