Method SetDeviceName

tensorflow/core/common_runtime/eager/eager_operation.cc:19–27

namespace tensorflow {

tensorflow::Status EagerOperation::SetDeviceName(const char* device) {
  if (device != nullptr && strlen(device) > 0) {
    if (!DeviceNameUtils::ParseFullName(device, &device_name_)) {
      return errors::InvalidArgument("Malformed device specification '", device,
                                     "' in eager op: ", DebugString());
    }
  }
  return Status::OK();
}

void EagerOperation::AddInput(tensorflow::TensorHandle* h) {
  h->Ref();