Home
last modified time | relevance | path

Searched refs:CustomDeviceOpHandler (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device_op_handler.cc22 void CustomDeviceOpHandler::Clear() { custom_devices_.clear(); } in Clear()
24 Status CustomDeviceOpHandler::RegisterCustomDevice( in RegisterCustomDevice()
44 bool CustomDeviceOpHandler::FindCustomDeviceFromName( in FindCustomDeviceFromName()
54 Status CustomDeviceOpHandler::Execute(ImmediateExecutionOperation* op, in Execute()
98 Status CustomDeviceOpHandler::MaybePinToCustomDevice( in MaybePinToCustomDevice()
Dcustom_device_op_handler.h25 class CustomDeviceOpHandler {
27 ~CustomDeviceOpHandler() {} in ~CustomDeviceOpHandler()
Dcontext.h482 CustomDeviceOpHandler& GetCustomDeviceOpHandler() override { in GetCustomDeviceOpHandler()
594 CustomDeviceOpHandler custom_device_op_handler_;
/external/tensorflow/tensorflow/c/eager/
Dimmediate_execution_context.h43 class CustomDeviceOpHandler; variable
157 virtual CustomDeviceOpHandler& GetCustomDeviceOpHandler() = 0;