Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py1811 use_cudnn_on_gpu=None, argument
2179 use_cudnn_on_gpu=True, argument
2302 use_cudnn_on_gpu=True, argument
2363 use_cudnn_on_gpu=True, argument
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad.cc219 bool use_cudnn_on_gpu; in Conv2DGrad() local
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc2677 bool use_cudnn_on_gpu; in CopyAttrsFromPadAndConv2D() local
2988 bool use_cudnn_on_gpu; in MergeConv2DWithBiasAdd() local