Searched refs:map_leaf (Results 1 – 1 of 1) sorted by relevance
46 std::shared_ptr<Dataset> map_leaf = ImageFolder("dir")->SetNumWorkers(0); in TEST_F() local48 std::shared_ptr<Dataset> batch = Zip({map_leaf, nonmap_leaf})->Batch(1)->SetNumWorkers(0); in TEST_F()51 EXPECT_EQ(map_leaf->IRNode()->NumWorkers(), 0); in TEST_F()63 EXPECT_NE(map_leaf->IRNode()->NumWorkers(), 0); in TEST_F()67 …MS_LOG(DEBUG) << map_leaf->IRNode()->Name() << ": num_worker=" << map_leaf->IRNode()->NumWorkers(); in TEST_F()