- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for getService (0.14 sec)
-
tensorflow/c/experimental/next_pluggable_device/c_api.cc
#include "tensorflow/core/platform/status.h" #include "tensorflow/core/tfrt/common/pjrt_util.h" #include "tsl/distributed_runtime/coordination/coordination_service_agent.h" TF_Device* TF_GetDevice(TF_OpKernelContext* ctx) { auto* cc_ctx = reinterpret_cast<tensorflow::OpKernelContext*>(ctx); return reinterpret_cast<TF_Device*>(cc_ctx->device()); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1)