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.py1545 use_cudnn_on_gpu=None, argument
1863 use_cudnn_on_gpu=True, argument
1961 use_cudnn_on_gpu=True, argument
2021 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/graph/
Dmkl_layout_pass.cc2095 bool use_cudnn_on_gpu; in CopyAttrsPadWithConv2D() local
2150 bool use_cudnn_on_gpu; in CopyAttrsFromPadAndConv2D() local
2684 bool use_cudnn_on_gpu; in MergeConv2DWithBiasAdd() local