- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for IsCustomDevice (0.12 sec)
-
tensorflow/c/eager/immediate_execution_context.h
virtual CustomDeviceOpHandler& GetCustomDeviceOpHandler() = 0; // Returns whether `device_name` is registered as a custom device. virtual bool IsCustomDevice(const string& device_name) = 0; // Register a custom device. It will return error is the device name is // already registered. // TODO(tfrt-devs): Remove this method. Let caller register it directly into
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 06 08:34:00 UTC 2023 - 12.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
return nullptr; } } // namespace extern "C" { bool TFE_IsCustomDevice(TFE_Context* ctx, const char* device_name) { return tensorflow::unwrap(ctx)->IsCustomDevice(device_name); } void TFE_RegisterCustomDevice(TFE_Context* ctx, TFE_CustomDevice device, const char* device_name, void* device_info,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0)