Home
last modified time | relevance | path

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

/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 …]
Dconstant_op_eager_test.py50 device = test_util.gpu_device_name()
Dcontrol_flow_ops_py_test.py331 with ops.device(test.gpu_device_name()):
366 with ops.device(test.gpu_device_name()):
529 with ops.device(test.gpu_device_name()):
1050 with ops.device(test.gpu_device_name()):
2883 gpu_dev_name = test.gpu_device_name() if test.is_gpu_available(
3392 with ops.device(test.gpu_device_name()):
4915 with ops.device(test.gpu_device_name()):
Drnn_cell_test.py2443 gpu_dev = test.gpu_device_name()
2466 gpu_dev = test.gpu_device_name()
2483 gpu_dev = test.gpu_device_name()
2905 gpu_dev = test.gpu_device_name()
Dfunctional_ops_test.py1278 with ops.device(test.gpu_device_name() if use_gpu else "CPU:0"):
/external/tensorflow/tensorflow/python/platform/
Dtest.py31 from tensorflow.python.framework.test_util import gpu_device_name
/external/tensorflow/tensorflow/python/profiler/internal/
Drun_metadata_test.py123 gpu_dev = test.gpu_device_name()
142 gpu_dev = test.gpu_device_name()
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc417 string gpu_device_name; in TensorHandleSilentCopy() local
418 if (GetDeviceName(ctx, &gpu_device_name, "GPU")) { in TensorHandleSilentCopy()
420 hcpu, ctx, gpu_device_name.c_str(), status.get()); in TensorHandleSilentCopy()
436 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status.get()); in TensorHandleSilentCopy()
489 string gpu_device_name; in SetAndGetOpDevices() local
490 if (GetDeviceName(ctx, &gpu_device_name, "GPU")) { in SetAndGetOpDevices()
567 string gpu_device_name; in TEST() local
568 if (GetDeviceName(ctx, &gpu_device_name, "GPU")) { in TEST()
570 hcpu, ctx, gpu_device_name.c_str(), status.get()); in TEST()
574 TFE_OpSetDevice(shape_op, gpu_device_name.c_str(), status.get()); in TEST()
[all …]
Dc_api_experimental_test.cc449 string gpu_device_name; in TEST() local
450 if (GetDeviceName(ctx, &gpu_device_name, "GPU")) { in TEST()
452 hcpu, ctx, gpu_device_name.c_str(), status.get()); in TEST()
456 TFE_OpSetDevice(shape_op, gpu_device_name.c_str(), status.get()); in TEST()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.test.pbtxt28 name: "gpu_device_name"
/external/tensorflow/tensorflow/python/kernel_tests/v1_compat_tests/
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()):
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.test.pbtxt44 name: "gpu_device_name"
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc2356 const string gpu_device_name = GPUDeviceName(session.get()); in TEST() local
2357 if (gpu_device_name.empty()) { in TEST()
2372 opts.mutable_fetch_devices()->insert({"y:0", gpu_device_name}); in TEST()
2389 opts.mutable_feed_devices()->insert({"x:0", gpu_device_name}); in TEST()
2428 const string gpu_device_name = GPUDeviceName(session.get()); in TestFeedAndFetchTensorsInDeviceMemory() local
2429 if (gpu_device_name.empty()) { in TestFeedAndFetchTensorsInDeviceMemory()
2447 opts.mutable_fetch_devices()->insert({"y:0", gpu_device_name}); in TestFeedAndFetchTensorsInDeviceMemory()
2464 opts.mutable_feed_devices()->insert({"x:0", gpu_device_name}); in TestFeedAndFetchTensorsInDeviceMemory()
2484 const string gpu_device_name = GPUDeviceName(session.get()); in TestFeedAndFetchTensorsInDeviceMemoryFailsToMakeCallable() local
2485 if (gpu_device_name.empty()) { in TestFeedAndFetchTensorsInDeviceMemoryFailsToMakeCallable()
[all …]
/external/tensorflow/tensorflow/python/ops/
Dmatmul_benchmark.py119 self.run_graph(test.gpu_device_name(), n, m, k, transpose_a, transpose_b,
Dstateful_random_ops_test.py144 with ops.device(test_util.gpu_device_name()):
451 self._sameAsOldRandomOps(test_util.gpu_device_name(), GPU_FLOATS)
463 with ops.device(test_util.gpu_device_name()):
Dimage_ops_test.py641 self._benchmarkFlipLeftRight(test.gpu_device_name(), None)
650 self._benchmarkRandomFlipLeftRight(test.gpu_device_name(), None)
659 self._benchmarkBatchedRandomFlipLeftRight(test.gpu_device_name(), None)
702 self._benchmarkAdjustHue(test.gpu_device_name(), None)
746 self._benchmarkAdjustSaturation(test.gpu_device_name(), None)
/external/tensorflow/tensorflow/python/eager/
Dcore_test.py1108 gpu_device_name = '/job:localhost/replica:0/task:0/device:GPU:0'
1114 self._recv(dtypes.float32, 't0', gpu_device_name),
1116 self._send(constant_op.constant(2.0), 't1', gpu_device_name)
/external/tensorflow/tensorflow/compiler/xla/g3doc/tutorials/
Dautoclustering_xla.ipynb66 "assert(tf.test.gpu_device_name())\n",
/external/tensorflow/tensorflow/python/debug/lib/
Ddumping_callback_test.py80 gpu_name = test_util.gpu_device_name()
1344 gpu_name = test_util.gpu_device_name()
Dsession_debug_testlib.py96 gpu_name = test_util.gpu_device_name()
763 if test_util.gpu_device_name():
/external/tensorflow/tensorflow/python/debug/cli/
Danalyzer_cli_test.py586 gpu_name = test_util.gpu_device_name()
1707 gpu_name = test_util.gpu_device_name()
/external/tensorflow/tensorflow/python/training/
Dsaver_test.py581 with sess.graph.device(test.gpu_device_name()):
588 with sess.graph.device(test.gpu_device_name()):
598 with sess.graph.device(test.gpu_device_name()):
605 with sess.graph.device(test.gpu_device_name()):
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py130 def gpu_device_name(): function
3202 gpu_name = gpu_device_name()
/external/tensorflow/tensorflow/c/
Dc_api_test.cc106 const string gpu_device_name = GPUDeviceName(sess); in GPUDeviceName() local
109 return gpu_device_name; in GPUDeviceName()