Searched refs:batch_element (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_utils_v1.py | 283 def create(self, batch_element): argument
284 self.composite = is_composite_or_composite_value(batch_element)
286 def aggregate(self, batch_element, batch_start=None, batch_end=None): argument
290 if self.batch_size and self.batch_size < batch_element.shape[0]:
294 batch_element.shape,
295 (self.batch_size,) + batch_element.shape[1:]))
296 self.results.append(batch_element)
369 def create(self, batch_element): argument
372 shape = (self.num_samples,) + batch_element.shape[1:]
373 dtype = batch_element.dtype
[all …]
|
| D | partial_batch_padding_handler.py | 85 for batch_element in dataset_batch_elements:
86 batch_elements.append(_pad(batch_element))
|
| D | training_utils_v1_test.py | 265 …def wrapped(batch_element, batch_start, batch_end, is_finished): # pylint: disable=unused-argument argument
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/benchmarks/ |
| D | index_lookup_forward_benchmark.py | 45 batch_element = []
48 batch_element.append(tok)
49 batch_element.append("") # Explicitly test the empty string.
50 data.append(batch_element)
|