OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:use_cudnn_on_gpu
(Results
1 - 6
of
6
) sorted by null
/external/tensorflow/tensorflow/cc/gradients/
nn_grad.cc
127
bool
use_cudnn_on_gpu
;
local
132
TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "
use_cudnn_on_gpu
", &
use_cudnn_on_gpu
));
135
input_attrs.UseCudnnOnGpu(
use_cudnn_on_gpu
);
142
filter_attrs.UseCudnnOnGpu(
use_cudnn_on_gpu
);
/external/tensorflow/tensorflow/python/ops/
nn_grad.py
53
use_cudnn_on_gpu
=op.get_attr("
use_cudnn_on_gpu
"),
61
use_cudnn_on_gpu
=op.get_attr("
use_cudnn_on_gpu
"),
76
use_cudnn_on_gpu
=op.get_attr("
use_cudnn_on_gpu
"),
84
use_cudnn_on_gpu
=op.get_attr("
use_cudnn_on_gpu
"),
502
use_cudnn_on_gpu
= op.get_attr("
use_cudnn_on_gpu
")
[
all
...]
nn_ops.py
[
all
...]
/external/tensorflow/tensorflow/contrib/quantize/python/
fold_batch_norms.py
386
use_cudnn_on_gpu
=layer_op.get_attr('
use_cudnn_on_gpu
'),
684
use_cudnn_on_gpu
=op.get_attr('
use_cudnn_on_gpu
'),
[
all
...]
/external/tensorflow/tensorflow/core/graph/
mkl_layout_pass.cc
1475
bool
use_cudnn_on_gpu
;
local
[
all
...]
/external/tensorflow/tensorflow/go/op/
wrappers.go
[
all
...]
Completed in 273 milliseconds