Searched refs:_generate_defun_backend (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_v2.py | 827 defun_standard_gru = _generate_defun_backend(api_name, _CPU_DEVICE_NAME,
830 defun_gpu_gru = _generate_defun_backend(api_name, _GPU_DEVICE_NAME,
1653 defun_standard_lstm = _generate_defun_backend(api_name, _CPU_DEVICE_NAME,
1656 defun_gpu_lstm = _generate_defun_backend(api_name, _GPU_DEVICE_NAME,
1744 def _generate_defun_backend(unique_api_name, preferred_device, func, function
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 64 def _generate_defun_backend(unique_api_name, preferred_device, func): function
917 ctc_loss_op_standard = _generate_defun_backend(api_name, _CPU_DEVICE_NAME,
919 ctc_loss_op_cudnn = _generate_defun_backend(api_name, _GPU_DEVICE_NAME,
|