Searched refs:new_batch_size (Results 1 – 1 of 1) sorted by relevance
576 const int64 new_batch_size = activations->shape().dimensions(batch_dimension); in IncreaseSpatialSizeOnSpaceToBatchedShape() local579 spatial_dim_size * new_batch_size / old_batch_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()582 << new_batch_size << " old_batch_size " << old_batch_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()594 ->set_edge_padding_high(new_spatial_dim_size * new_batch_size / in IncreaseSpatialSizeOnSpaceToBatchedShape()611 reshape_back_dims[batch_dimension] = new_batch_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()1421 const int64 new_batch_size = new_shape.dimensions(new_batch_dim); in Propagate() local1434 new_batch_dim, new_batch_size, in Propagate()1658 const int64 new_batch_size = new_shape.dimensions(new_batch_dim); in SelectValidPortion() local1662 CHECK_EQ(new_batch_size % old_batch_size, 0) in SelectValidPortion()1663 << " New batch size " << new_batch_size << " old batch size " in SelectValidPortion()[all …]