Searched defs:cudnn_use_autotune (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_ops_gpu.cc | 106 bool cudnn_use_autotune, in AutotuneFusedConv()
273 bool cudnn_use_autotune, in AutotuneUnfusedConv()
|
| H A D | conv_grad_input_ops.cc | 74 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|
| H A D | conv_grad_filter_ops.cc | 670 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|
| H A D | conv_ops.cc | 802 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|