1*a58d3d2aSXin Li /* Copyright (c) 2022 Amazon 2*a58d3d2aSXin Li Written by Jan Buethe */ 3*a58d3d2aSXin Li /* 4*a58d3d2aSXin Li Redistribution and use in source and binary forms, with or without 5*a58d3d2aSXin Li modification, are permitted provided that the following conditions 6*a58d3d2aSXin Li are met: 7*a58d3d2aSXin Li 8*a58d3d2aSXin Li - Redistributions of source code must retain the above copyright 9*a58d3d2aSXin Li notice, this list of conditions and the following disclaimer. 10*a58d3d2aSXin Li 11*a58d3d2aSXin Li - Redistributions in binary form must reproduce the above copyright 12*a58d3d2aSXin Li notice, this list of conditions and the following disclaimer in the 13*a58d3d2aSXin Li documentation and/or other materials provided with the distribution. 14*a58d3d2aSXin Li 15*a58d3d2aSXin Li THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 16*a58d3d2aSXin Li ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 17*a58d3d2aSXin Li LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 18*a58d3d2aSXin Li A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER 19*a58d3d2aSXin Li OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 20*a58d3d2aSXin Li EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 21*a58d3d2aSXin Li PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 22*a58d3d2aSXin Li PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 23*a58d3d2aSXin Li LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 24*a58d3d2aSXin Li NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 25*a58d3d2aSXin Li SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26*a58d3d2aSXin Li */ 27*a58d3d2aSXin Li 28*a58d3d2aSXin Li #ifndef DRED_RDOVAE_ENC_H 29*a58d3d2aSXin Li #define DRED_RDOVAE_ENC_H 30*a58d3d2aSXin Li 31*a58d3d2aSXin Li #include "dred_rdovae.h" 32*a58d3d2aSXin Li 33*a58d3d2aSXin Li #include "dred_rdovae_enc_data.h" 34*a58d3d2aSXin Li 35*a58d3d2aSXin Li struct RDOVAEEncStruct { 36*a58d3d2aSXin Li int initialized; 37*a58d3d2aSXin Li float gru1_state[ENC_GRU1_STATE_SIZE]; 38*a58d3d2aSXin Li float gru2_state[ENC_GRU2_STATE_SIZE]; 39*a58d3d2aSXin Li float gru3_state[ENC_GRU3_STATE_SIZE]; 40*a58d3d2aSXin Li float gru4_state[ENC_GRU4_STATE_SIZE]; 41*a58d3d2aSXin Li float gru5_state[ENC_GRU5_STATE_SIZE]; 42*a58d3d2aSXin Li float conv1_state[ENC_CONV1_STATE_SIZE]; 43*a58d3d2aSXin Li float conv2_state[2*ENC_CONV2_STATE_SIZE]; 44*a58d3d2aSXin Li float conv3_state[2*ENC_CONV3_STATE_SIZE]; 45*a58d3d2aSXin Li float conv4_state[2*ENC_CONV4_STATE_SIZE]; 46*a58d3d2aSXin Li float conv5_state[2*ENC_CONV5_STATE_SIZE]; 47*a58d3d2aSXin Li }; 48*a58d3d2aSXin Li 49*a58d3d2aSXin Li void dred_rdovae_encode_dframe(RDOVAEEncState *enc_state, const RDOVAEEnc *model, float *latents, float *initial_state, const float *input, int arch); 50*a58d3d2aSXin Li 51*a58d3d2aSXin Li 52*a58d3d2aSXin Li #endif 53