Searched refs:persistent_alloc_ids_by_devices_ (Results 1 – 2 of 2) sorted by relevance
373 if (persistent_alloc_ids_by_devices_.find(node->assigned_device_name()) == in IsPersistentTensor()374 persistent_alloc_ids_by_devices_.end()) { in IsPersistentTensor()377 return persistent_alloc_ids_by_devices_.at(node->assigned_device_name()) in IsPersistentTensor()
231 std::map<string, std::set<int64_t>> persistent_alloc_ids_by_devices_; variable