Searched defs:RegisterCustomDevice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ | ||
| D | custom_device_op_handler.cc | 24 Status CustomDeviceOpHandler::RegisterCustomDevice( in RegisterCustomDevice() function in tensorflow::CustomDeviceOpHandler |
| D | context.cc | 1140 Status EagerContext::RegisterCustomDevice( in RegisterCustomDevice() function in tensorflow::EagerContext |
| /external/tensorflow/tensorflow/core/tfrt/eager/ | ||
| D | c_api_tfrt.cc | 1130 tensorflow::Status ContextInterface::RegisterCustomDevice( in RegisterCustomDevice() function in tfrt::tf::ContextInterface |