- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for rumgr (0.13 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
} return pjrt_c_api_client; } absl::Status ResetPjRtClient(const DeviceType& device_type) { ResourceMgr* rmgr = tfrt_global::GetTFGlobalResourceMgr(); PjRtState* pjrt_state; TF_RETURN_IF_ERROR(rmgr->Lookup(rmgr->default_container(), kPjRtStateResourceName, &pjrt_state)); TF_RETURN_IF_ERROR(pjrt_state->MovePjRtClientToUnused(device_type));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 3.7K bytes - Viewed (0)