- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetCustomDeviceOpHandler (0.35 sec)
-
tensorflow/c/eager/c_api.cc
TFE_CustomDeviceTensorHandleMethods methods, TF_Status* status) { tensorflow::ImmediateExecutionContext* context = tensorflow::unwrap(ctx); tensorflow::CustomDevice* device = nullptr; if (!context->GetCustomDeviceOpHandler().FindCustomDeviceFromName(device_name, &device)) { methods.deallocator(data); status->status =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_context.h
// Experimental Custom Device. //===--------------------------------------------------------------------===// virtual CustomDeviceOpHandler& GetCustomDeviceOpHandler() = 0; // Returns whether `device_name` is registered as a custom device. virtual bool IsCustomDevice(const string& device_name) = 0;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 06 08:34:00 UTC 2023 - 12.3K bytes - Viewed (0)