Elytra does not function underwater. In Java, flying in the air and swooping down underwater will still make the elytra function. In bedrock, it just exits out of elytra mode right as it hits the water :/.