| 54 | } |
| 55 | |
| 56 | void TFE_TensorHandleCache::Insert(PyObject* value, tensorflow::DataType dtype, |
| 57 | absl::string_view device_name, |
| 58 | TFE_TensorHandle* handle) { |
| 59 | Py_INCREF(value); |
| 60 | handle->handle->Ref(); |
| 61 | cache.emplace(Key{PyObjectPtr{value}, dtype, device_name}, handle->handle); |
| 62 | } |
| 63 | |
| 64 | void TFE_TensorHandleCache::Clear() { |
| 65 | DecrefUnrefAll(); |
no test coverage detected