Searched refs:gpu_gru_with_fallback (Results 1 – 1 of 1) sorted by relevance
763 def gpu_gru_with_fallback(inputs, init_h, kernel, recurrent_kernel, bias, function815 _GPU_DEVICE_NAME: lambda: gpu_gru_with_fallback(**params)831 gpu_gru_with_fallback,