Searched refs:custom_device_op_handler_ (Results 1 – 3 of 3) sorted by relevance
114 if (custom_device_op_handler_.FindCustomDeviceFromName(device_name, &dev)) { in CopyTensorHandleToDevice()131 if (custom_device_op_handler_.FindCustomDeviceFromName(handle_device_name, in CopyTensorHandleToDevice()
483 return custom_device_op_handler_; in GetCustomDeviceOpHandler()594 CustomDeviceOpHandler custom_device_op_handler_; variable
525 custom_device_op_handler_.Clear(); in ~EagerContext()914 return custom_device_op_handler_.RegisterCustomDevice(device_name, in RegisterCustomDevice()