Searched defs:use_cudnn (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | depthwise_conv_op_d9m_test.py | 37 use_cudnn=False, argument
80 use_cudnn=False, argument
111 use_cudnn=False, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 74 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|
| D | depthwise_conv_op.cc | 429 bool use_cudnn = in Compute() local
|
| D | depthwise_conv_grad_op.cc | 644 bool use_cudnn = in Compute() local
1149 bool use_cudnn = std::is_same<Device, GPUDevice>::value && in Compute() local
|
| D | conv_grad_filter_ops.cc | 670 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|
| D | conv_ops.cc | 802 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 202 use_cudnn=False): argument
|