Searched refs:old_batch_size (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 169 HloInstruction* activations, int64 batch_dimension, int64 old_batch_size,
178 int64 activations_batch_dim, int64 old_batch_size, int64 low_padding,
325 const int64 old_batch_size = in IsConvSuitableForSpaceToBatch() local
328 if (old_batch_size > limit_on_batch_size_) { in IsConvSuitableForSpaceToBatch()
345 int64 activations_batch_dim, int64 old_batch_size, int64 low_padding, in HaloDuplicateWithSlice() argument
570 HloInstruction* activations, int64 batch_dimension, int64 old_batch_size, in IncreaseSpatialSizeOnSpaceToBatchedShape() argument
579 spatial_dim_size * new_batch_size / old_batch_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()
582 << new_batch_size << " old_batch_size " << old_batch_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()
584 new_dimensions[batch_dimension] = old_batch_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()
595 old_batch_size - in IncreaseSpatialSizeOnSpaceToBatchedShape()
[all …]
|