| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | rebatch_dataset_test.py | 42 def testBasic(self, drop_remainder): argument
61 def testNotDivisibleError(self, drop_remainder): argument
70 def testTupleOutput(self, drop_remainder): argument
80 def testNestedDictionaryOutput(self, drop_remainder): argument
90 def testFinalPartialBatchOriginal(self, drop_remainder): argument
104 def testFinalPartialBatchAfterRebatch(self, drop_remainder): argument
120 def testMultipleBatches(self, drop_remainder): argument
143 def testMapAndBatch(self, drop_remainder): argument
158 def testPaddedBatch(self, drop_remainder): argument
181 def testConcatenate(self, drop_remainder): argument
[all …]
|
| D | bucket_by_sequence_length_test.py | 419 def _compute_expected_batches(drop_remainder): argument
|
| D | map_and_batch_test.py | 134 def testMapAndBatchPartialBatch(self, drop_remainder, numa_aware): argument
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | numa_map_and_batch_dataset_serialization_test.py | 41 def build_ds(range_start, drop_remainder=False): argument
71 def build_ds(range_start, drop_remainder=False): argument
|
| D | map_and_batch_dataset_serialization_test.py | 42 def build_ds(range_start, drop_remainder=False): argument
69 def build_ds(range_start, drop_remainder=False): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 42 bool drop_remainder = false; in MakeDataset() local
54 Dataset(OpKernelContext* ctx, int64 batch_size, bool drop_remainder, in Dataset()
115 Node* drop_remainder = nullptr; in AsGraphDefInternal() local
|
| D | padded_batch_dataset_op.cc | 43 bool drop_remainder = false; in MakeDataset() local
103 Dataset(OpKernelContext* ctx, int64 batch_size, bool drop_remainder, in Dataset()
193 Node* drop_remainder = nullptr; in AsGraphDefInternal() local
|
| D | window_dataset_op.cc | 55 bool drop_remainder; in MakeDataset() local
67 int64 window_shift, int64 window_stride, bool drop_remainder) in Dataset()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | batch_test.py | 45 def testBatchDataset(self, count, batch_size, drop_remainder): argument
|
| D | padded_batch_test.py | 52 drop_remainder): argument
|
| D | window_test.py | 62 def testWindowDataset(self, count, size, shift, stride, drop_remainder=True): argument
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | batching.py | 235 drop_remainder=False, argument
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 569 drop_remainder, use_legacy_function=False): argument
617 drop_remainder=False, argument
677 drop_remainder=False, argument
|
| D | grouping.py | 134 drop_remainder=False): argument
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 848 def batch(self, batch_size, drop_remainder=False): argument
875 drop_remainder=False): argument
1197 def window(self, size, shift=None, stride=1, drop_remainder=False): argument
1647 def batch(self, batch_size, drop_remainder=False): argument
1656 drop_remainder=False): argument
1748 def window(self, size, shift=None, stride=1, drop_remainder=False): argument
2761 def __init__(self, input_dataset, batch_size, drop_remainder): argument
2908 drop_remainder): argument
3232 def __init__(self, input_dataset, size, shift, stride, drop_remainder): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | numa_map_and_batch_dataset_op.cc | 83 bool drop_remainder; in MakeDataset() local
102 int64 num_parallel_calls, bool drop_remainder, in Dataset()
516 Status GetBatch(IteratorContext* ctx, bool drop_remainder, in GetBatch()
|
| D | map_and_batch_dataset_op.cc | 84 bool drop_remainder; in MakeDataset() local
148 int64 num_parallel_calls, bool drop_remainder, in Dataset()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization_test.cc | 125 auto drop_remainder = graph_utils::AddScalarConstNode(true, graph); in AddBatchNode() local
291 auto drop_remainder = graph_utils::AddScalarConstNode(true, graph); in AddMapAndBatchNode() local
|