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_DEC_H 29*a58d3d2aSXin Li #define DRED_RDOVAE_DEC_H 30*a58d3d2aSXin Li 31*a58d3d2aSXin Li #include "dred_rdovae.h" 32*a58d3d2aSXin Li #include "dred_rdovae_dec_data.h" 33*a58d3d2aSXin Li #include "dred_rdovae_stats_data.h" 34*a58d3d2aSXin Li 35*a58d3d2aSXin Li struct RDOVAEDecStruct { 36*a58d3d2aSXin Li int initialized; 37*a58d3d2aSXin Li float gru1_state[DEC_GRU1_STATE_SIZE]; 38*a58d3d2aSXin Li float gru2_state[DEC_GRU2_STATE_SIZE]; 39*a58d3d2aSXin Li float gru3_state[DEC_GRU3_STATE_SIZE]; 40*a58d3d2aSXin Li float gru4_state[DEC_GRU4_STATE_SIZE]; 41*a58d3d2aSXin Li float gru5_state[DEC_GRU5_STATE_SIZE]; 42*a58d3d2aSXin Li float conv1_state[DEC_CONV1_STATE_SIZE]; 43*a58d3d2aSXin Li float conv2_state[DEC_CONV2_STATE_SIZE]; 44*a58d3d2aSXin Li float conv3_state[DEC_CONV3_STATE_SIZE]; 45*a58d3d2aSXin Li float conv4_state[DEC_CONV4_STATE_SIZE]; 46*a58d3d2aSXin Li float conv5_state[DEC_CONV5_STATE_SIZE]; 47*a58d3d2aSXin Li }; 48*a58d3d2aSXin Li 49*a58d3d2aSXin Li void dred_rdovae_dec_init_states(RDOVAEDecState *h, const RDOVAEDec *model, const float * initial_state, int arch); 50*a58d3d2aSXin Li void dred_rdovae_decode_qframe(RDOVAEDecState *h, const RDOVAEDec *model, float *qframe, const float * z, int arch); 51*a58d3d2aSXin Li void DRED_rdovae_decode_all(const RDOVAEDec *model, float *features, const float *state, const float *latents, int nb_latents, int arch); 52*a58d3d2aSXin Li 53*a58d3d2aSXin Li #endif 54