return absl::OkStatus();
        case 64:
          *tf_dtype = TF_DataType::TF_UINT64;
          return absl::OkStatus();
        default:
          return tensorflow::errors::InvalidArgument("Unsupported UInt bits: ",
                                                     dtype.bits);
      }
      return absl::OkStatus();
    case DLDataTypeCode::kDLInt:
      switch (dtype.bits) {
        case 8: