| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 271 void StallStreamOnError(LocalDeviceState* local_device, se::Stream* stream) { in StallStreamOnError() argument
272 switch (local_device->allocation_model()) { in StallStreamOnError()
282 if (stream != local_device->compute_stream()) { in StallStreamOnError()
283 local_device->compute_stream()->ThenWaitFor(stream); in StallStreamOnError()
365 LocalDeviceState* local_device, se::Stream* copy_stream, in AllocateDestinationBuffer() argument
378 local_device->device_ordinal())); in AllocateDestinationBuffer()
379 if (local_device->allocation_model() == in AllocateDestinationBuffer()
384 copy_stream->ThenWaitFor(local_device->compute_stream()); in AllocateDestinationBuffer()
388 local_device->compute_stream()->parent(), dst_buffer)); in AllocateDestinationBuffer()
397 if (local_device->allocation_model() == in AllocateDestinationBuffer()
[all …]
|
| D | gpu_device.cc | 105 for (auto& local_device : addressable_devices) { in CreateBFCAllocator() local
106 se::StreamExecutor* executor = local_device->executor(); in CreateBFCAllocator()
138 local_device->compute_stream()); in CreateBFCAllocator()
232 for (auto& local_device : local_device_states) { in BuildLocalDevices() local
233 int device_ordinal = local_device->device_ordinal(); in BuildLocalDevices()
235 local_device->executor()->GetDeviceDescription(); in BuildLocalDevices()
237 device_ordinal, std::move(local_device), description.name(), in BuildLocalDevices()
251 for (const auto& local_device : local_device_states) { in BuildDistributedDevices() local
252 const se::Platform* platform = local_device->executor()->platform(); in BuildDistributedDevices()
255 platform->DescriptionForDevice(local_device->device_ordinal())); in BuildDistributedDevices()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device.cc | 67 Device* local_device; in AsRemoteDevices() local
69 lookup_local_device(da.name(), &local_device).ok()) { in AsRemoteDevices()
71 local_device->name(), local_device, false, false)); in AsRemoteDevices()
|
| D | master_test.cc | 156 EXPECT_EQ("CPU", resp.local_device(0).device_type()); in TEST_F()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resources.py | 90 local_device = os.environ.get(
92 with ops.device(local_device):
|
| D | variables.py | 3416 local_device = os.environ.get(
3418 with ops.device(local_device):
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 943 Device* local_device; in TEST_F() local
944 TF_ASSERT_OK(device_mgr_->LookupDevice(local_device_, &local_device)); in TEST_F()
946 input_dev_ptrs.push_back(local_device); in TEST_F()
956 /*collective_executor=*/nullptr, local_device, fdef_.signature().name(), in TEST_F()
991 Device* local_device; in TEST_F() local
992 TF_ASSERT_OK(device_mgr_->LookupDevice(local_device_, &local_device)); in TEST_F()
994 input_dev_ptrs.push_back(local_device); in TEST_F()
1004 /*collective_executor=*/nullptr, local_device, fdef_.signature().name(), in TEST_F()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel_test.cc | 224 const string& local_device = "/job:localhost/replica:0/task:0/device:CPU:0"; in TEST_F() local
233 DeviceNameUtils::ParseFullName(local_device, &local_device_name); in TEST_F()
245 CreateNodeDef("OpWithoutKernel", {DT_STRING, DT_STRING}, local_device); in TEST_F()
248 DeviceNameUtils::ParseFullName(local_device, &local_device_name); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | BUILD | 237 "local_device.h",
524 ":local_device",
906 name = "local_device",
907 srcs = ["local_device.cc"],
908 hdrs = ["local_device.h"],
1536 ":local_device",
1593 ":local_device",
|
| D | placer_test.cc | 1590 const string local_device = "/job:b/replica:0/task:0/device:FakeCPU:0"; in TEST_F() local
1604 std::unique_ptr<Device> cpu(FakeDevice::MakeCPU(local_device)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session.cc | 394 for (const auto& device_attr : resp.local_device()) { in ListDevices()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 1587 for local_device in device_lib.list_local_devices():
1588 if local_device.device_type == "GPU":
1589 gpu_info = gpu_util.compute_capability_from_device_desc(local_device)
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | master.proto | 275 repeated DeviceAttributes local_device = 1; field
|
| /external/vulkan-validation-layers/tests/ |
| D | vklayertests_others.cpp | 1449 VkDevice local_device; in TEST_F() local
1461 VkResult err = vkCreateDevice(gpu(), &device_create_info, nullptr, &local_device); in TEST_F()
1471 vkCreateQueryPool(local_device, &qpci, nullptr, &query_pool); in TEST_F()
1474 vkDestroyDevice(local_device, nullptr); in TEST_F()
|