Home
last modified time | relevance | path

Searched refs:gpu_device_name (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/python/kernel_tests/
Dstage_op_test.py38 with ops.device(test.gpu_device_name()):
58 with ops.device(test.gpu_device_name()):
79 with ops.device(test.gpu_device_name()):
100 gpu_dev = test.gpu_device_name()
123 with ops.device(test.gpu_device_name()):
147 with ops.device(test.gpu_device_name()):
175 with ops.device(test.gpu_device_name()):
239 with ops.device(test.gpu_device_name()):
Dmap_stage_op_test.py41 with ops.device(test.gpu_device_name()):
63 with ops.device(test.gpu_device_name()):
86 with ops.device(test.gpu_device_name()):
107 gpu_dev = test.gpu_device_name()
136 with ops.device(test.gpu_device_name()):
166 with ops.device(test.gpu_device_name()):
194 with ops.device(test.gpu_device_name()):
260 with ops.device(test.gpu_device_name()):
324 with ops.device(test.gpu_device_name()):
362 with ops.device(test.gpu_device_name()):
[all …]
Dsession_ops_test.py152 with ops.device(test.gpu_device_name()):
186 with ops.device(test.gpu_device_name()):
212 with ops.device(test.gpu_device_name()):
Dconstant_op_eager_test.py50 device = test_util.gpu_device_name()
Dcontrol_flow_ops_py_test.py305 with ops.device(test.gpu_device_name()):
340 with ops.device(test.gpu_device_name()):
497 with ops.device(test.gpu_device_name()):
911 with ops.device(test.gpu_device_name()):
2499 gpu_dev_name = test.gpu_device_name() if test.is_gpu_available(
2950 with ops.device(test.gpu_device_name()):
4434 with ops.device(test.gpu_device_name()):
Drnn_cell_test.py2432 gpu_dev = test.gpu_device_name()
2454 gpu_dev = test.gpu_device_name()
2470 gpu_dev = test.gpu_device_name()
2895 gpu_dev = test.gpu_device_name()
/external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/
Drnn_ptb_graph_test.py35 with tf.Graph().as_default(), tf.device(tf.test.gpu_device_name()):
87 with tf.device(tf.test.gpu_device_name()):
100 tf.test.gpu_device_name(), PTBBenchmark.BATCH_SIZE)
130 with tf.device(tf.test.gpu_device_name()):
145 tf.test.gpu_device_name(), PTBBenchmark.BATCH_SIZE)
/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental_test.cc57 string gpu_device_name; in ExecuteWithProfiling() local
58 if (GetDeviceName(ctx, &gpu_device_name, "GPU")) { in ExecuteWithProfiling()
59 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status); in ExecuteWithProfiling()
81 if (!gpu_device_name.empty()) { in ExecuteWithProfiling()
Dc_api_test.cc667 string gpu_device_name; in TensorHandleSilentCopy() local
668 if (GetDeviceName(ctx, &gpu_device_name, "GPU")) { in TensorHandleSilentCopy()
670 hcpu, ctx, gpu_device_name.c_str(), status.get()); in TensorHandleSilentCopy()
674 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status.get()); in TensorHandleSilentCopy()
713 string gpu_device_name; in TensorHandleSilentCopyLocal() local
714 if (GetDeviceName(ctx, &gpu_device_name, "GPU")) { in TensorHandleSilentCopyLocal()
716 hcpu, ctx, gpu_device_name.c_str(), status.get()); in TensorHandleSilentCopyLocal()
720 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status.get()); in TensorHandleSilentCopyLocal()
753 string gpu_device_name; in SetAndGetOpDevices() local
754 if (GetDeviceName(ctx, &gpu_device_name, "GPU")) { in SetAndGetOpDevices()
[all …]
/external/tensorflow/tensorflow/python/ops/
Dmatmul_benchmark_test.py74 matmul_benchmark.build_graph(googletest.gpu_device_name(), n, m, k,
77 dev = googletest.gpu_device_name()
106 duration = benchmark_instance.run_graph(googletest.gpu_device_name(), n, m,
Dstateful_random_ops_test.py228 with ops.device(test_util.gpu_device_name()):
235 with ops.device(test_util.gpu_device_name()):
240 with ops.device(test_util.gpu_device_name()):
Dmatmul_benchmark.py119 self.run_graph(test.gpu_device_name(), n, m, k, transpose_a, transpose_b,
Dimage_ops_test.py581 self._benchmarkFlipLeftRight(test.gpu_device_name(), None)
590 self._benchmarkRandomFlipLeftRight(test.gpu_device_name(), None)
599 self._benchmarkBatchedRandomFlipLeftRight(test.gpu_device_name(), None)
642 self._benchmarkAdjustHue(test.gpu_device_name(), None)
686 self._benchmarkAdjustSaturation(test.gpu_device_name(), None)
/external/tensorflow/tensorflow/python/profiler/internal/
Drun_metadata_test.py119 gpu_dev = test.gpu_device_name()
135 gpu_dev = test.gpu_device_name()
/external/tensorflow/tensorflow/python/platform/
Dtest.py37 from tensorflow.python.framework.test_util import gpu_device_name
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc1927 const string gpu_device_name = GPUDeviceName(session.get()); in TEST() local
1928 if (gpu_device_name.empty()) { in TEST()
1943 opts.mutable_fetch_devices()->insert({"y:0", gpu_device_name}); in TEST()
1960 opts.mutable_feed_devices()->insert({"x:0", gpu_device_name}); in TEST()
1999 const string gpu_device_name = GPUDeviceName(session.get()); in TestFeedAndFetchTensorsInDeviceMemory() local
2000 if (gpu_device_name.empty()) { in TestFeedAndFetchTensorsInDeviceMemory()
2018 opts.mutable_fetch_devices()->insert({"y:0", gpu_device_name}); in TestFeedAndFetchTensorsInDeviceMemory()
2035 opts.mutable_feed_devices()->insert({"x:0", gpu_device_name}); in TestFeedAndFetchTensorsInDeviceMemory()
2055 const string gpu_device_name = GPUDeviceName(session.get()); in TestFeedAndFetchTensorsInDeviceMemoryFailsToMakeCallable() local
2056 if (gpu_device_name.empty()) { in TestFeedAndFetchTensorsInDeviceMemoryFailsToMakeCallable()
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.test.pbtxt28 name: "gpu_device_name"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.test.pbtxt44 name: "gpu_device_name"
/external/tensorflow/tensorflow/contrib/image/python/kernel_tests/
Ddistort_image_ops_test.py295 self._benchmark_adjust_hue_in_yiq(test.gpu_device_name(), None)
340 self._benchmark_adjust_saturation_in_yiq(test.gpu_device_name(), None)
/external/tensorflow/tensorflow/contrib/eager/python/
Ddatasets_test.py191 with ops.device(test.gpu_device_name()):
198 with ops.device(test.gpu_device_name()):
/external/tensorflow/tensorflow/python/eager/
Dcore_test.py745 gpu_device_name = '/job:localhost/replica:0/task:0/device:GPU:0'
751 self._recv(dtypes.float32, 't0', gpu_device_name),
753 self._send(constant_op.constant(2.0), 't1', gpu_device_name)
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py99 def gpu_device_name(): function
2547 gpu_name = gpu_device_name()
/external/tensorflow/tensorflow/python/debug/cli/
Danalyzer_cli_test.py586 gpu_name = test_util.gpu_device_name()
1681 gpu_name = test_util.gpu_device_name()
/external/tensorflow/tensorflow/python/debug/lib/
Dsession_debug_testlib.py96 gpu_name = test_util.gpu_device_name()
765 if test_util.gpu_device_name():
/external/tensorflow/tensorflow/python/training/
Dsaver_test.py582 with sess.graph.device(test.gpu_device_name()):
589 with sess.graph.device(test.gpu_device_name()):
599 with sess.graph.device(test.gpu_device_name()):
606 with sess.graph.device(test.gpu_device_name()):

12