Searched defs:cudnn_use_autotune (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 544 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|
| D | conv_grad_input_ops.cc | 681 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|
| D | conv_ops.cc | 579 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 441 launch(OpKernelContext* ctx, bool cudnn_use_autotune, in launch()
|