parameters: (seq_len, embed_dim, hidden_dim, latent_dim)
return_statement: return flops
docstring: Estimate FLOPs/sample in ATOM VAE encoder forward prop
docstring_summary: Estimate FLOPs/sample in ATOM VAE encoder forward prop
docstring_tokens: Estimate FLOPs / sample in ATOM VAE encoder forward prop
function: def encoder_fp_flops(seq_len, embed_dim, hidden_dim, latent_dim):
      """Estimate FLOPs/sample in ATOM VAE encoder forward prop"""
      flops = 0
  
      # RNN
      flops += 2 * (3*hidden_dim*embed_dim + 3*hidden_dim**2) * seq_len
      flops += 2 * (3*hidden_dim) * seq_len
  
      # FC
      flops += 2 * (latent_dim*hidden_dim) + latent_dim
  
      return flops
function_tokens: def encoder_fp_flops ( seq_len , embed_dim , hidden_dim , latent_dim ) : flops = 0 # RNN flops += 2 * ( 3 * hidden_dim * embed_dim + 3 * hidden_dim ** 2 ) * seq_len flops += 2 * ( 3 * hidden_dim ) * seq_len # FC flops += 2 * ( latent_dim * hidden_dim ) + latent_dim return flops