Searched refs:GetLocalDeviceState (Results 1 – 2 of 2) sorted by relevance
127 StatusOr<LocalDeviceState*> PjRtStreamExecutorDevice::GetLocalDeviceState() in GetLocalDeviceState() function in xla::PjRtStreamExecutorDevice660 ->GetLocalDeviceState()); in BufferFromHostBuffer()843 ->GetLocalDeviceState()); in CreateUninitializedBuffer()867 ->GetLocalDeviceState()); in BufferFromHostLiteral()931 ->GetLocalDeviceState(); in MakeCrossHostReceiveBuffers()974 TF_ASSIGN_OR_RETURN(LocalDeviceState * local_device, GetLocalDeviceState()); in TransferToInfeed()981 TF_ASSIGN_OR_RETURN(LocalDeviceState * local_device, GetLocalDeviceState()); in TransferFromOutfeed()1377 ->GetLocalDeviceState()); in CopyToDevice()
102 StatusOr<LocalDeviceState*> GetLocalDeviceState() const;