Home
last modified time | relevance | path

Searched refs:use_cudnn_on_gpu (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dnn_grad.py52 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
60 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
75 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
83 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
571 use_cudnn_on_gpu = op.get_attr("use_cudnn_on_gpu")
590 use_cudnn_on_gpu=use_cudnn_on_gpu,
600 use_cudnn_on_gpu=use_cudnn_on_gpu,
Dnn_ops.py1545 use_cudnn_on_gpu=None, argument
1620 use_cudnn_on_gpu=use_cudnn_on_gpu,
1686 use_cudnn_on_gpu=True,
1851 use_cudnn_on_gpu=True,
1863 use_cudnn_on_gpu=True, argument
1947 use_cudnn_on_gpu=use_cudnn_on_gpu,
1961 use_cudnn_on_gpu=True, argument
2010 input, filter_sizes, out_backprop, strides, padding, use_cudnn_on_gpu,
2021 use_cudnn_on_gpu=True, argument
2074 input_sizes, filter, out_backprop, strides, padding, use_cudnn_on_gpu,
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad.cc219 bool use_cudnn_on_gpu; in Conv2DGrad() local
224 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "use_cudnn_on_gpu", &use_cudnn_on_gpu)); in Conv2DGrad()
228 .UseCudnnOnGpu(use_cudnn_on_gpu)); in Conv2DGrad()
234 .UseCudnnOnGpu(use_cudnn_on_gpu)); in Conv2DGrad()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt69 …argspec: "args=[\'value\', \'filters\', \'stride\', \'padding\', \'use_cudnn_on_gpu\', \'data_form…
77 …argspec: "args=[\'input\', \'filter\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'data_form…
81 …\', \'filter_sizes\', \'out_backprop\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'data_for…
85 …_sizes\', \'filter\', \'out_backprop\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'data_for…
Dtensorflow.raw_ops.pbtxt661 …argspec: "args=[\'input\', \'filter\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit_…
665 …\', \'filter_sizes\', \'out_backprop\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit…
669 …_sizes\', \'filter\', \'out_backprop\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit…
/external/tensorflow/tensorflow/core/profiler/internal/testdata/
Dgraph.pbtxt870 key: "use_cudnn_on_gpu"
1355 key: "use_cudnn_on_gpu"
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc2095 bool use_cudnn_on_gpu; in CopyAttrsPadWithConv2D() local
2104 GetNodeAttr(orig_node->def(), "use_cudnn_on_gpu", &use_cudnn_on_gpu)); in CopyAttrsPadWithConv2D()
2117 nb->Attr("use_cudnn_on_gpu", use_cudnn_on_gpu); in CopyAttrsPadWithConv2D()
2150 bool use_cudnn_on_gpu; in CopyAttrsFromPadAndConv2D() local
2159 GetNodeAttr(orig_node1->def(), "use_cudnn_on_gpu", &use_cudnn_on_gpu)); in CopyAttrsFromPadAndConv2D()
2169 nb->Attr("use_cudnn_on_gpu", use_cudnn_on_gpu); in CopyAttrsFromPadAndConv2D()
2684 bool use_cudnn_on_gpu; in MergeConv2DWithBiasAdd() local
2692 TF_CHECK_OK(GetNodeAttr(pred->def(), "use_cudnn_on_gpu", &use_cudnn_on_gpu)); in MergeConv2DWithBiasAdd()
/external/tensorflow/tensorflow/examples/tutorials/deepdream/
Ddeepdream.ipynb243 …use_cudnn_on_gpu"\\n value {\\n b: true\\n }\\n }\\n}\\nnode {\\n name: "co…
675 …use_cudnn_on_gpu"\\n value {\\n b: true\\n }\\n }\\n}\\nnode {\\n name: "sp…
/external/tensorflow/tensorflow/contrib/quantize/python/
Dfold_batch_norms.py456 use_cudnn_on_gpu=layer_op.get_attr('use_cudnn_on_gpu'),
869 use_cudnn_on_gpu=op.get_attr('use_cudnn_on_gpu'),
/external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/
Dlarge_function_graph.pbtxt533 key: "use_cudnn_on_gpu"
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py1593 use_cudnn_on_gpu = pfor_input.get_attr("use_cudnn_on_gpu")
1609 use_cudnn_on_gpu=use_cudnn_on_gpu,
1635 use_cudnn_on_gpu=use_cudnn_on_gpu,
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt7901 name: "use_cudnn_on_gpu"
7947 name: "use_cudnn_on_gpu"
8007 name: "use_cudnn_on_gpu"
8070 name: "use_cudnn_on_gpu"
8120 name: "use_cudnn_on_gpu"
8184 name: "use_cudnn_on_gpu"
8247 name: "use_cudnn_on_gpu"
8297 name: "use_cudnn_on_gpu"
8361 name: "use_cudnn_on_gpu"
Dops_history.v2.pbtxt13954 name: "use_cudnn_on_gpu"
14014 name: "use_cudnn_on_gpu"
14087 name: "use_cudnn_on_gpu"
14160 name: "use_cudnn_on_gpu"
14244 name: "use_cudnn_on_gpu"
14308 name: "use_cudnn_on_gpu"
14385 name: "use_cudnn_on_gpu"
14462 name: "use_cudnn_on_gpu"
14546 name: "use_cudnn_on_gpu"
14610 name: "use_cudnn_on_gpu"
[all …]
Dops_history.v1.pbtxt14012 name: "use_cudnn_on_gpu"
14072 name: "use_cudnn_on_gpu"
14145 name: "use_cudnn_on_gpu"
14218 name: "use_cudnn_on_gpu"
14302 name: "use_cudnn_on_gpu"
14366 name: "use_cudnn_on_gpu"
14443 name: "use_cudnn_on_gpu"
14520 name: "use_cudnn_on_gpu"
14604 name: "use_cudnn_on_gpu"
14668 name: "use_cudnn_on_gpu"
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt661 …argspec: "args=[\'input\', \'filter\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit_…
665 …\', \'filter_sizes\', \'out_backprop\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit…
669 …_sizes\', \'filter\', \'out_backprop\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt6226 name: "use_cudnn_on_gpu"
6312 name: "use_cudnn_on_gpu"
6398 name: "use_cudnn_on_gpu"