Home
last modified time | relevance | path

Searched defs:use_cudnn_on_gpu (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py1993 use_cudnn_on_gpu=None, argument
2368 use_cudnn_on_gpu=True, argument
2491 use_cudnn_on_gpu=True, argument
2552 use_cudnn_on_gpu=True, argument
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Dnn_grad.cc219 bool use_cudnn_on_gpu; in Conv2DGrad() local
524 bool use_cudnn_on_gpu; in Conv2DBackpropInputGrad() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc2654 bool use_cudnn_on_gpu; in CopyAttrsFromPadAndConv2D() local
2952 bool use_cudnn_on_gpu; in MergeConv2DWithBiasAdd() local