H A D | recurrent_v2.py | 541 def standard_gru(inputs, init_h, kernel, recurrent_kernel, bias, mask, argument 625 def gpu_gru(inputs, init_h, kernel, recurrent_kernel, bias, mask, time_major, argument 711 def gru_with_backend_selection(inputs, init_h, kernel, recurrent_kernel, bias, argument 761 def gpu_gru_with_fallback(inputs, init_h, kernel, recurrent_kernel, bias, argument 1322 def standard_lstm(inputs, init_h, init_c, kernel, recurrent_kernel, bias, argument 1407 def gpu_lstm(inputs, init_h, init_c, kernel, recurrent_kernel, bias, mask, argument 1532 def lstm_with_backend_selection(inputs, init_h, init_c, kernel, argument 1585 def gpu_lstm_with_fallback(inputs, init_h, init_c, kernel, recurrent_kernel, argument
|