Searched refs:device_restrictions (Results 1 – 1 of 1) sorted by relevance
239 std::unordered_map<string, std::set<string>> device_restrictions; in TF_GetSupportedDevices() local263 device_restrictions[tensorflow::grappler::NodeName(node.input(i))].insert("CPU"); in TF_GetSupportedDevices()279 const auto it = device_restrictions.find(node); in TF_GetSupportedDevices()280 if (it != device_restrictions.end()) { in TF_GetSupportedDevices()