Searched defs:local_device (Results 1 – 8 of 8) sorted by relevance
274 void StallStreamOnError(LocalDeviceState* local_device, se::Stream* stream) { in StallStreamOnError()375 LocalDeviceState* local_device, se::Stream* copy_stream, in AllocateDestinationBuffer()490 LocalDeviceState* local_device, in AddDestinationBufferSynchronization()595 auto* local_device = device_->local_device_state(); in logical_on_device_shape() local1319 LocalDeviceState* local_device = device_->local_device_state(); in ToLiteral() local1675 PjRtStreamExecutorClient* client, LocalDeviceState* local_device, in MakeTupleHelper()1747 PjRtDevice* device, LocalDeviceState* local_device, in OutputBufferHelper()
366 std::unique_ptr<LocalDeviceState> local_device; in BuildDistributedDevices() local
67 Device* local_device; in AsRemoteDevices() local
275 repeated DeviceAttributes local_device = 1; field
949 Device* local_device; in TEST_F() local1003 Device* local_device; in TEST_F() local
227 const string& local_device = "/job:localhost/replica:0/task:0/device:CPU:0"; in TEST_F() local
1602 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