Searched refs:num_datasets (Results 1 – 5 of 5) sorted by relevance
77 int64 num_datasets = 10; in TEST_P() local79 for (int i = 0; i < num_datasets; ++i) { in TEST_P()85 for (int i = 0; i < num_datasets; ++i) { in TEST_P()
175 num_datasets = len(datasets)179 logits = [[1.0] * num_datasets]188 if not weights.shape.is_compatible_with([num_datasets]):
63 def _testSampleFromDatasetsHelper(self, weights, num_datasets, num_samples): argument68 for i in range(num_datasets)73 freqs = np.zeros([num_datasets])
136 num_datasets = 3139 for _ in range(num_datasets):145 for dataset_ind in range(num_datasets):
539 const int64 num_datasets = dataset()->datasets_.size(); in GetNextInternal() local540 *end_of_sequence = num_datasets == index_; in GetNextInternal()