- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for RegisterCustomDevice (0.19 sec)
-
tensorflow/c/eager/custom_device_testutil.cc
device->executed_flag = executed_flag; device->device_name = name; device->underlying_device = "/job:localhost/replica:0/task:0/device:CPU:0"; device->strict_scope_placement = strict_scope_placement; TFE_RegisterCustomDevice(context, custom_device, name, device, status); } TFE_TensorHandle* UnpackTensorHandle(TFE_TensorHandle* logged_tensor_handle, TF_Status* status) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
} } // 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, TF_Status* status) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2)