- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetCustomDeviceOpHandler (0.15 sec)
-
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: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 12.3K bytes - Viewed (0) -
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: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0)