Home
last modified time | relevance | path

Searched refs:ReceiveRawFromOutfeed (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Doutfeed_receiver.cc191 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