Searched defs:local_device (Results 1 – 8 of 8) sorted by relevance
105 for (auto& local_device : addressable_devices) { in CreateBFCAllocator() local232 for (auto& local_device : local_device_states) { in BuildLocalDevices() local251 for (const auto& local_device : local_device_states) { in BuildDistributedDevices() local274 std::unique_ptr<LocalDeviceState> local_device; in BuildDistributedDevices() local
271 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()936 LocalDeviceState* local_device = local_device_or.ConsumeValueOrDie(); in MakeCrossHostReceiveBuffers() local1218 LocalDeviceState* local_device = in ToLiteral() local1481 PjRtClient* client, LocalDeviceState* local_device, in MakeTupleHelper()1553 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() local991 Device* local_device; in TEST_F() 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