Searched defs:local_device (Results 1 – 9 of 9) sorted by relevance
245 for (auto& local_device : addressable_devices) { in CreateBFCAllocator() local397 for (auto& local_device : local_device_states) { in BuildLocalDevices() local417 for (const auto& local_device : local_device_states) { in BuildDistributedDevices() local440 std::unique_ptr<LocalDeviceState> local_device; in BuildDistributedDevices() local
264 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() local1005 LocalDeviceState* local_device = local_device_or.ConsumeValueOrDie(); in MakeCrossHostReceiveBuffers() local1053 LocalDeviceState* local_device = local_device_or.ConsumeValueOrDie(); in MakeCrossHostReceiveBuffersForGather() local1320 LocalDeviceState* local_device = device_->local_device_state(); in ToLiteral() local1629 PjRtStreamExecutorClient* client, LocalDeviceState* local_device, in MakeTupleHelper()1701 PjRtDevice* device, LocalDeviceState* local_device, in OutputBufferHelper()
67 Device* local_device; in AsRemoteDevices() local
275 repeated DeviceAttributes local_device = 1; field
943 Device* local_device; in TEST_F() local992 Device* local_device; in TEST_F() local
141 for (const auto& local_device : local_devices) { in GenerateIncarnationIdTensor() local
224 const string& local_device = "/job:localhost/replica:0/task:0/device:CPU:0"; in TEST_F() local
1590 const string local_device = "/job:b/replica:0/task:0/device:FakeCPU:0"; in TEST_F() local
1449 VkDevice local_device; in TEST_F() local