Searched defs:use_cudnn_on_gpu (Results 1 – 3 of 3) sorted by relevance
219 bool use_cudnn_on_gpu; in Conv2DGrad() local524 bool use_cudnn_on_gpu; in Conv2DBackpropInputGrad() local
1931 use_cudnn_on_gpu=None, argument2299 use_cudnn_on_gpu=True, argument2422 use_cudnn_on_gpu=True, argument2483 use_cudnn_on_gpu=True, argument
2690 bool use_cudnn_on_gpu; in CopyAttrsFromPadAndConv2D() local2988 bool use_cudnn_on_gpu; in MergeConv2DWithBiasAdd() local