| D | recurrent_v2.py | 543 def standard_gru(inputs, init_h, kernel, recurrent_kernel, bias, mask, argument
615 inputs, [init_h],
627 def gpu_gru(inputs, init_h, kernel, recurrent_kernel, bias, mask, time_major, argument
637 init_h = array_ops.expand_dims(init_h, axis=seq_axis)
677 input_h=init_h,
693 input=inputs, input_h=init_h, input_c=0, params=params,
713 def gru_with_backend_selection(inputs, init_h, kernel, recurrent_kernel, bias, argument
752 'init_h': init_h,
763 def gpu_gru_with_fallback(inputs, init_h, kernel, recurrent_kernel, bias, argument
770 init_h=init_h,
[all …]
|