| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | rebatch_dataset_test.py | 42 def testBasic(self, drop_remainder): argument
44 32, drop_remainder=drop_remainder)
47 [[32 if drop_remainder else None]],
50 [[8 if drop_remainder else None]],
61 def testNotDivisibleError(self, drop_remainder): argument
63 32, drop_remainder=drop_remainder)
70 def testTupleOutput(self, drop_remainder): argument
73 32, drop_remainder=drop_remainder))
80 def testNestedDictionaryOutput(self, drop_remainder): argument
83 32, drop_remainder=drop_remainder)
[all …]
|
| D | cardinality_test.py | 36 lambda: dataset_ops.Dataset.range(5).batch(2, drop_remainder=True), 2),
38 lambda: dataset_ops.Dataset.range(5).batch(2, drop_remainder=False), 3),
93 2, [], drop_remainder=True), 2),
95 2, [], drop_remainder=False), 3),
139 size=2, shift=2, drop_remainder=True), 2),
141 size=2, shift=2, drop_remainder=False), 3),
|
| D | map_and_batch_test.py | 134 def testMapAndBatchPartialBatch(self, drop_remainder, numa_aware): argument
140 drop_remainder=drop_remainder)))
147 if drop_remainder:
154 if not drop_remainder:
217 lambda x: x, batch_size=100, drop_remainder=True))
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | map_and_batch_dataset_serialization_test.py | 42 def build_ds(range_start, drop_remainder=False): argument
53 drop_remainder=drop_remainder))
69 def build_ds(range_start, drop_remainder=False): argument
80 drop_remainder=drop_remainder))
|
| D | numa_map_and_batch_dataset_serialization_test.py | 41 def build_ds(range_start, drop_remainder=False): argument
52 drop_remainder=drop_remainder))
71 def build_ds(range_start, drop_remainder=False): argument
82 drop_remainder=drop_remainder))
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 42 bool drop_remainder = false; in MakeDataset() local
45 &drop_remainder)); in MakeDataset()
48 *output = new Dataset(ctx, batch_size, drop_remainder, input); in MakeDataset()
54 Dataset(OpKernelContext* ctx, int64 batch_size, bool drop_remainder, in Dataset() argument
58 drop_remainder_(drop_remainder), in Dataset()
115 Node* drop_remainder = nullptr; in AsGraphDefInternal() local
116 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
118 this, {input_graph_node, batch_size, drop_remainder}, output)); in AsGraphDefInternal()
|
| D | padded_batch_dataset_op.cc | 43 bool drop_remainder = false; in MakeDataset() local
46 &drop_remainder)); in MakeDataset()
96 new Dataset(ctx, batch_size, drop_remainder, std::move(padded_shapes), in MakeDataset()
103 Dataset(OpKernelContext* ctx, int64 batch_size, bool drop_remainder, in Dataset() argument
108 drop_remainder_(drop_remainder), in Dataset()
193 Node* drop_remainder = nullptr; in AsGraphDefInternal() local
194 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
203 this, {{0, input_graph_node}, {1, batch_size}, {4, drop_remainder}}, in AsGraphDefInternal()
|
| D | window_dataset_op.cc | 55 bool drop_remainder; in MakeDataset() local
57 ctx, ParseScalarArgument<bool>(ctx, "drop_remainder", &drop_remainder)); in MakeDataset()
60 drop_remainder); in MakeDataset()
67 int64 window_shift, int64 window_stride, bool drop_remainder) in Dataset() argument
73 drop_remainder_(drop_remainder) { in Dataset()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | window_test.py | 62 def testWindowDataset(self, count, size, shift, stride, drop_remainder=True): argument
81 drop_remainder=drop_remainder).flat_map(_flat_map_fn)
96 if not drop_remainder:
133 drop_remainder=True).flat_map(lambda x: x.batch(batch_size=5))
155 drop_remainder=True).flat_map(lambda x: x.batch(batch_size=5))
181 drop_remainder=True).flat_map(lambda x: x.batch(batch_size=4)).window(
183 drop_remainder=True).flat_map(lambda x: x.batch(batch_size=3))
223 drop_remainder=True).flat_map(lambda x: x.batch(batch_size=2))
|
| D | batch_test.py | 45 def testBatchDataset(self, count, batch_size, drop_remainder): argument
65 _map_fn).repeat(count).batch(batch_size, drop_remainder)
68 if drop_remainder:
84 if not drop_remainder and (count * 7) % batch_size > 0:
|
| D | padded_batch_test.py | 52 drop_remainder): argument
66 drop_remainder=drop_remainder,
82 if not drop_remainder and len(seq_lens) % batch_size > 0:
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | batching.py | 133 return dataset.batch(batch_size, drop_remainder=True)
170 drop_remainder=True)
235 drop_remainder=False, argument
273 drop_remainder, num_parallel_calls)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 520 drop_remainder = d._drop_remainder
523 drop_remainder = d._drop_remainder_t
529 if tensor_util.is_tensor(drop_remainder):
530 drop_remainder = tensor_util.constant_value(drop_remainder)
532 return batch_size, drop_remainder
544 batch_size, drop_remainder = _get_batched_dataset_attributes(batched_dataset)
554 return batch_size, drop_remainder, prefetch_buffer
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | grouping.py | 134 drop_remainder=False): argument
217 return grouped_dataset.batch(batch_size, drop_remainder=drop_remainder)
236 batch_size, shapes, padding_values, drop_remainder=drop_remainder)
|
| D | batching.py | 569 drop_remainder, use_legacy_function=False): argument
583 drop_remainder, dtype=dtypes.bool, name="drop_remainder")
599 drop_remainder=self._drop_remainder_t,
617 drop_remainder=False, argument
662 num_parallel_calls, drop_remainder,
677 drop_remainder=False, argument
726 num_parallel_calls, drop_remainder)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.-dataset.pbtxt | 15 …argspec: "args=[\'self\', \'batch_size\', \'drop_remainder\'], varargs=None, keywords=None, defaul…
63 …=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\'], varargs=No…
99 …argspec: "args=[\'self\', \'size\', \'shift\', \'stride\', \'drop_remainder\'], varargs=None, keyw…
|
| D | tensorflow.data.experimental.-random-dataset.pbtxt | 17 …argspec: "args=[\'self\', \'batch_size\', \'drop_remainder\'], varargs=None, keywords=None, defaul…
65 …=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\'], varargs=No…
101 …argspec: "args=[\'self\', \'size\', \'shift\', \'stride\', \'drop_remainder\'], varargs=None, keyw…
|
| D | tensorflow.data.-text-line-dataset.pbtxt | 17 …argspec: "args=[\'self\', \'batch_size\', \'drop_remainder\'], varargs=None, keywords=None, defaul…
65 …=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\'], varargs=No…
101 …argspec: "args=[\'self\', \'size\', \'shift\', \'stride\', \'drop_remainder\'], varargs=None, keyw…
|
| D | tensorflow.data.-t-f-record-dataset.pbtxt | 16 …argspec: "args=[\'self\', \'batch_size\', \'drop_remainder\'], varargs=None, keywords=None, defaul…
64 …=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\'], varargs=No…
100 …argspec: "args=[\'self\', \'size\', \'shift\', \'stride\', \'drop_remainder\'], varargs=None, keyw…
|
| D | tensorflow.data.experimental.-sql-dataset.pbtxt | 17 …argspec: "args=[\'self\', \'batch_size\', \'drop_remainder\'], varargs=None, keywords=None, defaul…
65 …=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\'], varargs=No…
101 …argspec: "args=[\'self\', \'size\', \'shift\', \'stride\', \'drop_remainder\'], varargs=None, keyw…
|
| D | tensorflow.data.-fixed-length-record-dataset.pbtxt | 17 …argspec: "args=[\'self\', \'batch_size\', \'drop_remainder\'], varargs=None, keywords=None, defaul…
65 …=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\'], varargs=No…
101 …argspec: "args=[\'self\', \'size\', \'shift\', \'stride\', \'drop_remainder\'], varargs=None, keyw…
|
| D | tensorflow.data.experimental.-csv-dataset.pbtxt | 17 …argspec: "args=[\'self\', \'batch_size\', \'drop_remainder\'], varargs=None, keywords=None, defaul…
65 …=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\'], varargs=No…
101 …argspec: "args=[\'self\', \'size\', \'shift\', \'stride\', \'drop_remainder\'], varargs=None, keyw…
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 848 def batch(self, batch_size, drop_remainder=False): argument
869 return BatchDataset(self, batch_size, drop_remainder)
875 drop_remainder=False): argument
927 drop_remainder)
1197 def window(self, size, shift=None, stride=1, drop_remainder=False): argument
1234 return WindowDataset(self, size, shift, stride, drop_remainder)
1647 def batch(self, batch_size, drop_remainder=False): argument
1649 batch_size, drop_remainder))
1656 drop_remainder=False): argument
1658 batch_size, padded_shapes, padding_values, drop_remainder))
[all …]
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | metrics_v1_test.py | 40 4, drop_remainder=True)
53 3, drop_remainder=True)
66 3, drop_remainder=True)
141 4, drop_remainder=True)
206 dataset = dataset.batch(4, drop_remainder=True)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.-dataset.pbtxt | 28 …argspec: "args=[\'self\', \'batch_size\', \'drop_remainder\'], varargs=None, keywords=None, defaul…
96 …=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\'], varargs=No…
132 …argspec: "args=[\'self\', \'size\', \'shift\', \'stride\', \'drop_remainder\'], varargs=None, keyw…
|