Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/engine/
Dpartial_batch_padding_handler.py34 self.padded_batch_size = 0
59 missing_count = self.padded_batch_size - original_batch_size
76 missing_count = (self.padded_batch_size -
Dtraining_distributed.py607 padding_handler.padded_batch_size = batch_size
/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc336 const int padded_batch_size = RoundToLowestAllowedBatchSize(batch.size()); in ConcatInputTensors() local
337 const int padding_amount = padded_batch_size - batch.size(); in ConcatInputTensors()