Searched refs:FindCustomDeviceFromName (Results 1 – 4 of 4) sorted by relevance
44 bool CustomDeviceOpHandler::FindCustomDeviceFromName( in FindCustomDeviceFromName() function in tensorflow::CustomDeviceOpHandler101 if (!FindCustomDeviceFromName(op.DeviceName(), device) && in MaybePinToCustomDevice()
33 bool FindCustomDeviceFromName(const string& name,
114 if (custom_device_op_handler_.FindCustomDeviceFromName(device_name, &dev)) { in CopyTensorHandleToDevice()131 if (custom_device_op_handler_.FindCustomDeviceFromName(handle_device_name, in CopyTensorHandleToDevice()
537 if (!context->GetCustomDeviceOpHandler().FindCustomDeviceFromName(device_name, in TFE_NewCustomDeviceTensorHandle()568 if (!context->GetCustomDeviceOpHandler().FindCustomDeviceFromName( in TFE_NewTensorHandleFromDeviceMemory()