Searched refs:bucket_batch_sizes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | grouping.py | 99 bucket_batch_sizes, argument
142 element_length_func, bucket_boundaries, bucket_batch_sizes, padded_shapes,
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | grouping.py | 129 bucket_batch_sizes, argument
177 if len(bucket_batch_sizes) != (len(bucket_boundaries) + 1):
181 batch_sizes = constant_op.constant(bucket_batch_sizes, dtype=dtypes.int64)
224 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 | 89 …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],
|