Searched refs:_GPU_DEVICE_NAME (Results 1 – 2 of 2) sorted by relevance
52 _GPU_DEVICE_NAME = 'GPU' variable526 (device_type == _GPU_DEVICE_NAME or815 _GPU_DEVICE_NAME: lambda: gpu_gru_with_fallback(**params)830 defun_gpu_gru = _generate_defun_backend(api_name, _GPU_DEVICE_NAME,1254 (device_type == _GPU_DEVICE_NAME or1641 _GPU_DEVICE_NAME: lambda: gpu_lstm_with_fallback(**params)1656 defun_gpu_lstm = _generate_defun_backend(api_name, _GPU_DEVICE_NAME,
53 _GPU_DEVICE_NAME = "GPU" variable908 (device_type == _GPU_DEVICE_NAME or919 ctc_loss_op_cudnn = _generate_defun_backend(api_name, _GPU_DEVICE_NAME,