Searched refs:ReceiveRawFromOutfeed (Results 1 – 1 of 1) sorted by relevance
191 StatusOr<std::unique_ptr<Literal>> ReceiveRawFromOutfeed(PjRtDevice* device,291 ReceiveRawFromOutfeed(device, header_shape).ValueOrDie(); in DeviceListenerThreadLoop()323 ReceiveRawFromOutfeed(device, shape).ValueOrDie(); in DeviceListenerThreadLoop()342 StatusOr<std::unique_ptr<Literal>> OutfeedReceiverImpl::ReceiveRawFromOutfeed( in ReceiveRawFromOutfeed() function in xla::OutfeedReceiverImpl