Searched refs:TFE_RegisterCustomDevice (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_testlib.h | 162 TFE_RegisterCustomDevice(context, device, device_name, device_info, status); in RegisterParallelDevice()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | custom_device_testutil.cc | 168 TFE_RegisterCustomDevice(context, custom_device, name, device, status); in RegisterLoggingDevice()
|
| D | c_api_experimental.h | 514 TF_CAPI_EXPORT extern void TFE_RegisterCustomDevice(TFE_Context* ctx,
|
| D | c_api.cc | 1146 void TFE_RegisterCustomDevice(TFE_Context* ctx, TFE_CustomDevice device, in TFE_RegisterCustomDevice() function
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 1448 TFE_RegisterCustomDevice( in PYBIND11_MODULE()
|