| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | rebatch_dataset_test.py | 178 def testShapeInferenceInputBatchDimDivisible(self, drop_remainder): argument
227 def testBasic(self, drop_remainder): argument
265 def testBatchSizeGreaterThanOriginal(self, drop_remainder): argument
277 def testEmptySplits(self, drop_remainder): argument
294 def testEmptyFirstSplits(self, drop_remainder): argument
313 def testEmptyLastSplits(self, drop_remainder): argument
328 def testScalarBatchSizeInput(self, drop_remainder): argument
370 def testRaggedDataset(self, drop_remainder): argument
406 def testBasic(self, drop_remainder): argument
456 def testBatchNotDivisibleByNumReplicas(self, drop_remainder): argument
[all …]
|
| D | map_and_batch_test.py | 126 def testMapAndBatchPartialBatch(self, drop_remainder): argument
430 def testNumParallelBatches(self, verify_fn, drop_remainder): argument
442 def build_ds(range_start, drop_remainder): argument
464 def testNumParallelCalls(self, verify_fn, drop_remainder): argument
476 def build_ds(range_start, drop_remainder=False): argument
|
| D | dense_to_ragged_batch_test.py | 116 def testBasic(self, make_dataset, nrows, batch_size, drop_remainder): argument
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 37 drop_remainder=False, argument
156 drop_remainder=False, argument
216 drop_remainder=False, argument
331 drop_remainder, use_legacy_function=False): argument
|
| D | grouping.py | 123 drop_remainder=False): argument
|
| D | distribute.py | 125 def __init__(self, input_dataset, batch_sizes, drop_remainder=False): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 50 Dataset(OpKernelContext* ctx, int64_t batch_size, bool drop_remainder, in Dataset()
137 Node* drop_remainder = nullptr; in AsGraphDefInternal() local
272 bool drop_remainder = false; in MakeDataset() local
|
| D | padded_batch_dataset_op.cc | 52 Dataset(OpKernelContext* ctx, int64_t batch_size, bool drop_remainder, in Dataset()
162 Node* drop_remainder = nullptr; in AsGraphDefInternal() local
434 bool drop_remainder = false; in MakeDataset() local
|
| D | parallel_batch_dataset_op_test.cc | 30 int64_t num_parallel_calls, bool drop_remainder, in ParallelBatchDatasetParams()
53 Tensor drop_remainder = in GetInputTensors() local
|
| D | parallel_batch_dataset_op.cc | 72 bool drop_remainder, bool parallel_copy, const DatasetBase* input, in Dataset()
165 Node* drop_remainder = nullptr; in AsGraphDefInternal() local
648 bool drop_remainder = false; in MakeDataset() local
|
| D | window_dataset_op.cc | 47 int64_t window_shift, int64_t window_stride, bool drop_remainder) in Dataset()
414 bool drop_remainder; in MakeDataset() local
|
| D | window_dataset_op_test.cc | 31 int64_t stride, bool drop_remainder, in WindowDatasetParams()
|
| D | padded_batch_dataset_op_test.cc | 31 bool drop_remainder, bool parallel_copy, in PaddedBatchDatasetParams()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib_test.py | 602 drop_remainder, distribution): argument
632 iteration_type, drop_remainder, argument
693 drop_remainder, argument
1055 def testRaggedSparse(self, distribution, input_type, drop_remainder, argument
1178 drop_remainder, tensor_type, argument
1230 drop_remainder): argument
|
| D | input_lib_type_spec_test.py | 130 drop_remainder): argument
180 drop_remainder): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | rebatch_dataset_op.cc | 308 bool drop_remainder; in MakeDataset() local
320 std::vector<int64>&& batch_sizes, bool drop_remainder, in Dataset()
373 Node* drop_remainder = nullptr; in AsGraphDefInternal() local
|
| D | map_and_batch_dataset_op.cc | 85 int64_t num_parallel_calls, bool drop_remainder, in Dataset()
729 bool drop_remainder; in MakeDataset() local
|
| D | map_and_batch_dataset_op_test.cc | 28 int64_t batch_size, int64_t num_parallel_calls, bool drop_remainder, in MapAndBatchDatasetParams()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 1529 drop_remainder=False, argument
1593 drop_remainder=False): argument
2090 def window(self, size, shift=None, stride=1, drop_remainder=False): argument
2624 drop_remainder=False): argument
3384 drop_remainder=False, argument
3396 drop_remainder=False): argument
3508 def window(self, size, shift=None, stride=1, drop_remainder=False): argument
4546 def __init__(self, input_dataset, batch_size, drop_remainder): argument
4583 def __init__(self, input_dataset, batch_size, drop_remainder, argument
4812 drop_remainder): argument
[all …]
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | window_test.py | 53 def testWindowDataset(self, count, size, shift, stride, drop_remainder): argument
|
| D | bucket_by_sequence_length_test.py | 428 def _compute_expected_batches(drop_remainder): argument
|
| D | padded_batch_test.py | 50 def testPaddedBatchDataset(self, count, padded_shapes, drop_remainder): argument
|
| D | batch_test.py | 58 def testBasic(self, count, batch_size, drop_remainder, num_parallel_calls): argument
|
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset_test_base.h | 218 bool drop_remainder, bool parallel_copy, in BatchDatasetParams()
|
| D | dataset_utils.cc | 679 bool drop_remainder, const Status& status, in ProcessBatch()
|