/external/tensorflow/tensorflow/python/kernel_tests/ |
D | stage_op_test.py | 38 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()):
|
D | map_stage_op_test.py | 41 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 …]
|
D | session_ops_test.py | 152 with ops.device(test.gpu_device_name()): 186 with ops.device(test.gpu_device_name()): 212 with ops.device(test.gpu_device_name()):
|
D | constant_op_eager_test.py | 50 device = test_util.gpu_device_name()
|
D | control_flow_ops_py_test.py | 305 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()):
|
D | rnn_cell_test.py | 2432 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/ |
D | rnn_ptb_graph_test.py | 35 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/ |
D | c_api_experimental_test.cc | 57 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()
|
D | c_api_test.cc | 667 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/ |
D | matmul_benchmark_test.py | 74 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,
|
D | stateful_random_ops_test.py | 228 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()):
|
D | matmul_benchmark.py | 119 self.run_graph(test.gpu_device_name(), n, m, k, transpose_a, transpose_b,
|
D | image_ops_test.py | 581 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/ |
D | run_metadata_test.py | 119 gpu_dev = test.gpu_device_name() 135 gpu_dev = test.gpu_device_name()
|
/external/tensorflow/tensorflow/python/platform/ |
D | test.py | 37 from tensorflow.python.framework.test_util import gpu_device_name
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | direct_session_test.cc | 1927 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/ |
D | tensorflow.test.pbtxt | 28 name: "gpu_device_name"
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.test.pbtxt | 44 name: "gpu_device_name"
|
/external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
D | distort_image_ops_test.py | 295 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/ |
D | datasets_test.py | 191 with ops.device(test.gpu_device_name()): 198 with ops.device(test.gpu_device_name()):
|
/external/tensorflow/tensorflow/python/eager/ |
D | core_test.py | 745 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/ |
D | test_util.py | 99 def gpu_device_name(): function 2547 gpu_name = gpu_device_name()
|
/external/tensorflow/tensorflow/python/debug/cli/ |
D | analyzer_cli_test.py | 586 gpu_name = test_util.gpu_device_name() 1681 gpu_name = test_util.gpu_device_name()
|
/external/tensorflow/tensorflow/python/debug/lib/ |
D | session_debug_testlib.py | 96 gpu_name = test_util.gpu_device_name() 765 if test_util.gpu_device_name():
|
/external/tensorflow/tensorflow/python/training/ |
D | saver_test.py | 582 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()):
|