- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TF_GetDeviceName (0.12 sec)
-
tensorflow/c/kernels.h
// Returns the Device Name of the device that the context possesses. // // The returned TF_StringView's underlying string is owned by the OpKernel and // has the same lifetime as the OpKernel. TF_CAPI_EXPORT TF_StringView TF_GetDeviceName(TF_OpKernelContext* ctx); #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD) // Returns the rendezvous in the context. Not supported on mobile.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 22:46:22 UTC 2024 - 24.6K bytes - Viewed (0) -
tensorflow/c/kernels.cc
device->parsed_name(), tensorflow::DeviceType(device->device_type())); if (!id.ok()) return -1; return *id; #endif // defined(IS_MOBILE_PLATFORM) || defined(IS_SLIM_BUILD) } TF_StringView TF_GetDeviceName(TF_OpKernelContext* ctx) { const auto& device_name = reinterpret_cast<tensorflow::OpKernelContext*>(ctx)->device()->name(); TF_StringView device_name_sv; device_name_sv.data = device_name.data();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0)