Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dfused_kernel_matcher.mlir9 … fused_ops = ["BiasAdd"], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (te…
12 …, explicit_paddings = [], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (te…
20 …ps = ["BiasAdd", "Relu"], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (te…
23 …, explicit_paddings = [], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (te…
32 …s = ["BiasAdd", "Relu6"], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (te…
35 …, explicit_paddings = [], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (te…
44 …ops = ["BiasAdd", "Elu"], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (te…
47 …, explicit_paddings = [], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (te…
57 …, explicit_paddings = [], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (te…
67 … fused_ops = ["BiasAdd"], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (te…
[all …]
Dtpu_space_to_depth_pass.mlir59 … explicit_paddings = [], padding = "VALID", strides = [1, 2, 2, 1], use_cudnn_on_gpu = true} : (te…
63 … explicit_paddings = [], padding = "VALID", strides = [1, 2, 2, 1], use_cudnn_on_gpu = true} : (te…
136 … explicit_paddings = [], padding = "VALID", strides = [1, 2, 2, 1], use_cudnn_on_gpu = true} : (te…
210 … explicit_paddings = [], padding = "VALID", strides = [1, 2, 2, 1], use_cudnn_on_gpu = true} : (te…
Dtf_optimize.mlir15 …, 3, 1], explicit_paddings = [], padding = "SAME", strides = [1, 4, 5, 1], use_cudnn_on_gpu = true}
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DConv2D.pbtxt30 name: "use_cudnn_on_gpu"
90 name: "use_cudnn_on_gpu"
163 name: "use_cudnn_on_gpu"
236 name: "use_cudnn_on_gpu"
319 name: "use_cudnn_on_gpu"
DConv2DBackpropInput.pbtxt34 name: "use_cudnn_on_gpu"
98 name: "use_cudnn_on_gpu"
175 name: "use_cudnn_on_gpu"
252 name: "use_cudnn_on_gpu"
339 name: "use_cudnn_on_gpu"
DConv2DBackpropFilter.pbtxt34 name: "use_cudnn_on_gpu"
98 name: "use_cudnn_on_gpu"
175 name: "use_cudnn_on_gpu"
252 name: "use_cudnn_on_gpu"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DConv2D.pbtxt30 name: "use_cudnn_on_gpu"
90 name: "use_cudnn_on_gpu"
163 name: "use_cudnn_on_gpu"
236 name: "use_cudnn_on_gpu"
319 name: "use_cudnn_on_gpu"
DConv2DBackpropInput.pbtxt34 name: "use_cudnn_on_gpu"
98 name: "use_cudnn_on_gpu"
175 name: "use_cudnn_on_gpu"
252 name: "use_cudnn_on_gpu"
339 name: "use_cudnn_on_gpu"
DConv2DBackpropFilter.pbtxt34 name: "use_cudnn_on_gpu"
98 name: "use_cudnn_on_gpu"
175 name: "use_cudnn_on_gpu"
252 name: "use_cudnn_on_gpu"
/external/tensorflow/tensorflow/python/ops/
Dnn_grad.py53 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
62 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
80 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
89 use_cudnn_on_gpu=op.get_attr("use_cudnn_on_gpu"),
576 use_cudnn_on_gpu = op.get_attr("use_cudnn_on_gpu")
595 use_cudnn_on_gpu=use_cudnn_on_gpu,
605 use_cudnn_on_gpu=use_cudnn_on_gpu,
Dnn_ops.py1811 use_cudnn_on_gpu=None, argument
1889 use_cudnn_on_gpu=use_cudnn_on_gpu,
1901 use_cudnn_on_gpu=use_cudnn_on_gpu,
1972 use_cudnn_on_gpu=True,
2166 use_cudnn_on_gpu=True,
2179 use_cudnn_on_gpu=True, argument
2274 use_cudnn_on_gpu=use_cudnn_on_gpu,
2286 use_cudnn_on_gpu=use_cudnn_on_gpu,
2302 use_cudnn_on_gpu=True, argument
2351 input, filter_sizes, out_backprop, strides, padding, use_cudnn_on_gpu,
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
Dfuse-bias-tf.mlir12 …, explicit_paddings = [], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (te…
Dtf-to-tosa-pipeline.mlir14 …, explicit_paddings = [], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (t…
38 …, explicit_paddings = [], padding = "SAME", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (t…
/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/compiler/mlir/lite/tests/end2end/
Dconv_2d.pbtxt137 key: "use_cudnn_on_gpu"
Dconv_2d_nchw.pbtxt137 key: "use_cudnn_on_gpu"
/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…
/external/tensorflow/tensorflow/compiler/jit/tests/
Dkeras_imagenet_main_graph_mode.pbtxt28436 key: "use_cudnn_on_gpu"
30108 key: "use_cudnn_on_gpu"
30168 key: "use_cudnn_on_gpu"
30553 key: "use_cudnn_on_gpu"
30846 key: "use_cudnn_on_gpu"
31152 key: "use_cudnn_on_gpu"
31393 key: "use_cudnn_on_gpu"
31660 key: "use_cudnn_on_gpu"
31966 key: "use_cudnn_on_gpu"
32207 key: "use_cudnn_on_gpu"
[all …]
Dkeras_imagenet_main.pbtxt28909 key: "use_cudnn_on_gpu"
35078 key: "use_cudnn_on_gpu"
35138 key: "use_cudnn_on_gpu"
35476 key: "use_cudnn_on_gpu"
35746 key: "use_cudnn_on_gpu"
36029 key: "use_cudnn_on_gpu"
36247 key: "use_cudnn_on_gpu"
36491 key: "use_cudnn_on_gpu"
36774 key: "use_cudnn_on_gpu"
36992 key: "use_cudnn_on_gpu"
[all …]
/external/tensorflow/tensorflow/core/profiler/internal/testdata/
Dgraph.pbtxt870 key: "use_cudnn_on_gpu"
1355 key: "use_cudnn_on_gpu"
/external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/
Dlarge_function_graph.pbtxt542 key: "use_cudnn_on_gpu"
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/debuginfo/
Dv1_1.0_224_frozen.wrong_attr.line.part.pbtxt132 key: "use_cudnn_on_gpu"
Dv1_1.0_224_frozen.wrong_attr.stack.part.pbtxt135 key: "use_cudnn_on_gpu"
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc2677 bool use_cudnn_on_gpu; in CopyAttrsFromPadAndConv2D() local
2686 GetNodeAttr(orig_node1->def(), "use_cudnn_on_gpu", &use_cudnn_on_gpu)); in CopyAttrsFromPadAndConv2D()
2696 nb->Attr("use_cudnn_on_gpu", use_cudnn_on_gpu); in CopyAttrsFromPadAndConv2D()
2988 bool use_cudnn_on_gpu; in MergeConv2DWithBiasAdd() local
2996 TF_CHECK_OK(GetNodeAttr(pred->def(), "use_cudnn_on_gpu", &use_cudnn_on_gpu)); in MergeConv2DWithBiasAdd()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc368 strides, backprop.use_cudnn_on_gpu(), backprop.padding(), in HandleConv2DBackPropFilter()

12