Searched refs:component_ids (Results 1 – 2 of 2) sorted by relevance
195 std::unordered_map<int, std::vector<const NodeDef*>> component_ids; in IdentifyLoops() local201 component_ids[id].push_back(it.first); in IdentifyLoops()205 for (const auto& component : component_ids) { in IdentifyLoops()
268 typename TTypes<OutputType>::Flat component_ids, in operator()271 component_ids.device(device) = in operator()272 component_ids.generate(FindRootGenerator(images, union_find)); in operator()