- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for RegisterCustomDevice (0.25 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_testlib.h
TFE_CustomDevice device; void* device_info; tensorflow::parallel_device::AllocateParallelDevice( device_name, underlying_devices.data(), underlying_devices.size(), &device, &device_info); TFE_RegisterCustomDevice(context, device, device_name, device_info, status); } } // namespace parallel_device } // namespace tensorflow
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0)