Searched refs:OutfeedData (Results 1 – 1 of 1) sorted by relevance
103 class OutfeedData { class105 OutfeedData(PjRtDevice* device, uint32_t consumer_id, Shape shape) in OutfeedData() function in xla::OutfeedData134 void OutfeedData::SetLiteral(std::unique_ptr<Literal> literal) { in SetLiteral()147 std::string OutfeedData::DebugString() const { in DebugString()195 void EnqueueReceivedData(std::unique_ptr<OutfeedData> received)222 std::queue<std::unique_ptr<OutfeedData>> callback_queue_ TF_GUARDED_BY(mu_);309 auto received = absl::make_unique<OutfeedData>(device, consumer_id, shape); in DeviceListenerThreadLoop()331 std::unique_ptr<OutfeedData> received) TF_EXCLUSIVE_LOCKS_REQUIRED(mu_) { in EnqueueReceivedData()356 std::unique_ptr<OutfeedData> received; in CallbackThreadLoop()