Searched defs:batch_sizes (Results 1 – 3 of 3) sorted by relevance
296 std::vector<int64> batch_sizes; in MakeDataset() local314 std::vector<int64>&& batch_sizes, bool drop_remainder, in Dataset()359 Node* batch_sizes = nullptr; in AsGraphDefInternal() local
117 def _hvp_benchmark(self, hvp_fn, label, batch_sizes, argument
120 def __init__(self, input_dataset, batch_sizes, drop_remainder=False): argument