Home
last modified time | relevance | path

Searched defs:local_device (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc245 for (auto& local_device : addressable_devices) { in CreateBFCAllocator() local
397 for (auto& local_device : local_device_states) { in BuildLocalDevices() local
417 for (const auto& local_device : local_device_states) { in BuildDistributedDevices() local
440 std::unique_ptr<LocalDeviceState> local_device; in BuildDistributedDevices() local
Dpjrt_stream_executor_client.cc264 void StallStreamOnError(LocalDeviceState* local_device, se::Stream* stream) { in StallStreamOnError()
365 LocalDeviceState* local_device, se::Stream* copy_stream, in AllocateDestinationBuffer()
480 LocalDeviceState* local_device, in AddDestinationBufferSynchronization()
585 auto* local_device = device_->local_device_state(); in logical_on_device_shape() local
1005 LocalDeviceState* local_device = local_device_or.ConsumeValueOrDie(); in MakeCrossHostReceiveBuffers() local
1053 LocalDeviceState* local_device = local_device_or.ConsumeValueOrDie(); in MakeCrossHostReceiveBuffersForGather() local
1320 LocalDeviceState* local_device = device_->local_device_state(); in ToLiteral() local
1629 PjRtStreamExecutorClient* client, LocalDeviceState* local_device, in MakeTupleHelper()
1701 PjRtDevice* device, LocalDeviceState* local_device, in OutputBufferHelper()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dremote_device.cc67 Device* local_device; in AsRemoteDevices() local
/external/tensorflow/tensorflow/core/protobuf/
Dmaster.proto275 repeated DeviceAttributes local_device = 1; field
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc943 Device* local_device; in TEST_F() local
992 Device* local_device; in TEST_F() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext_distributed_manager.cc141 for (const auto& local_device : local_devices) { in GenerateIncarnationIdTensor() local
/external/tensorflow/tensorflow/core/framework/
Dop_kernel_test.cc224 const string& local_device = "/job:localhost/replica:0/task:0/device:CPU:0"; in TEST_F() local
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_test.cc1590 const string local_device = "/job:b/replica:0/task:0/device:FakeCPU:0"; in TEST_F() local
/external/vulkan-validation-layers/tests/
Dvklayertests_others.cpp1449 VkDevice local_device; in TEST_F() local