Searched refs:client_dataset (Results 1 – 5 of 5) sorted by relevance
18 Dataset::ClientDataset client_dataset; in TEST_F() local20 TestExampleIterator iterator(&client_dataset); in TEST_F()35 Dataset::ClientDataset client_dataset; in TEST_F() local36 client_dataset.add_example("abc"); in TEST_F()38 TestExampleIterator iterator(&client_dataset); in TEST_F()55 Dataset::ClientDataset client_dataset; in TEST_F() local56 client_dataset.add_example("a"); in TEST_F()57 client_dataset.add_example("b"); in TEST_F()58 client_dataset.add_example("c"); in TEST_F()60 TestExampleIterator iterator(&client_dataset); in TEST_F()
224 Dataset::ClientDataset client_dataset; in Initialize() local225 client_dataset.set_client_id("client_id"); in Initialize()226 client_dataset.add_example(example); in Initialize()227 dataset_.mutable_client_data()->Add(std::move(client_dataset)); in Initialize()428 Dataset::ClientDataset client_dataset; in TEST_F() local429 client_dataset.set_client_id("client_id"); in TEST_F()430 client_dataset.add_example(example); in TEST_F()432 dataset_.mutable_client_data()->Add(std::move(client_dataset)); in TEST_F()521 Dataset::ClientDataset client_dataset; in TEST_F() local522 client_dataset.add_example(invalid_example); in TEST_F()[all …]
113 for (const Dataset::ClientDataset& client_dataset : in InitializeFlTask() local115 num_examples_ += client_dataset.example_size(); in InitializeFlTask()116 for (const std::string& example : client_dataset.example()) { in InitializeFlTask()157 for (const Dataset::ClientDataset& client_dataset : in ComputeDatasetStats() local160 client_dataset.selected_example()) { in ComputeDatasetStats()
70 for (const Dataset::ClientDataset& client_dataset : dataset.client_data()) { in SimpleExampleIterator() local71 FCP_CHECK(client_dataset.selected_example_size() == 0) in SimpleExampleIterator()74 for (const auto& example : client_dataset.example()) { in SimpleExampleIterator()86 for (const Dataset::ClientDataset& client_dataset : dataset.client_data()) { in SimpleExampleIterator() local87 FCP_CHECK(client_dataset.selected_example_size() > 0) in SimpleExampleIterator()91 client_dataset.selected_example()) { in SimpleExampleIterator()
724 client_dataset = dataset_ops.Dataset.zip((targets, handles)).map(map_fn)725 itr = dataset_ops.make_initializable_iterator(client_dataset)