Searched refs:bucket_batch_sizes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | grouping.py | 130 bucket_batch_sizes, argument
203 if len(bucket_batch_sizes) != (len(bucket_boundaries) + 1):
207 batch_sizes = constant_op.constant(bucket_batch_sizes, dtype=dtypes.int64)
250 constant_op.constant(len(bucket_batch_sizes) - 1,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 89 …argspec: "args=[\'element_length_func\', \'bucket_boundaries\', \'bucket_batch_sizes\', \'padded_s…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 117 …argspec: "args=[\'element_length_func\', \'bucket_boundaries\', \'bucket_batch_sizes\', \'padded_s…
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | bucket_by_sequence_length_test.py | 375 bucket_batch_sizes=[2, 2, 2],
|
| D | data_service_ops_test.py | 336 bucket_batch_sizes = [batch_size] * (len(bucket_boundaries) + 1)
345 bucket_batch_sizes,
|