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.py1585 use_cudnn_on_gpu=None, argument
1926 use_cudnn_on_gpu=True, argument
2024 use_cudnn_on_gpu=True, argument
2084 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.cc2528 bool use_cudnn_on_gpu; in CopyAttrsPadWithConv2D() local
2583 bool use_cudnn_on_gpu; in CopyAttrsFromPadAndConv2D() local
2933 bool use_cudnn_on_gpu; in MergeConv2DWithBiasAdd() local