Home
last modified time | relevance | path

Searched defs:batch_sizes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drebatch_dataset_op.cc302 std::vector<int64_t> batch_sizes; in MakeDataset() local
320 std::vector<int64_t>&& batch_sizes, bool drop_remainder, in Dataset()
371 Node* batch_sizes = nullptr; in AsGraphDefInternal() local
Dauto_shard_dataset_op_test.cc338 Node* batch_sizes; in TEST_F() local
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dhvp_test.py112 def _hvp_benchmark(self, hvp_fn, label, batch_sizes, argument
/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddistribute.py121 def __init__(self, input_dataset, batch_sizes, drop_remainder=False): argument