Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Doutfeed_receiver_test.cc127 ->AddOutfeedToBuilder(&builder, CreateToken(&builder), in TEST()
160 ->AddOutfeedToBuilder(&builder0, CreateToken(&builder0), in TEST()
170 ->AddOutfeedToBuilder(&builder1, CreateToken(&builder1), in TEST()
205 ->AddOutfeedToBuilder(&builder, CreateToken(&builder), in TEST()
214 ->AddOutfeedToBuilder(&builder, send0, consumer_id1, {data1}) in TEST()
248 ->AddOutfeedToBuilder(&builder, CreateToken(&builder), in TEST()
255 StatusOr<XlaOp> send1 = outfeed_receiver->AddOutfeedToBuilder( in TEST()
280 StatusOr<XlaOp> send0 = outfeed_receiver->AddOutfeedToBuilder( in TEST()
308 ->AddOutfeedToBuilder(&builder, CreateToken(&builder), in TEST()
Doutfeed_receiver.cc167 StatusOr<XlaOp> AddOutfeedToBuilder(XlaBuilder* builder, XlaOp token,
395 AddOutfeedToBuilder(&builder, CreateToken(&builder), consumer_id, {}) in SendShutdownOutfeedHeader()
417 StatusOr<XlaOp> OutfeedReceiverImpl::AddOutfeedToBuilder( in AddOutfeedToBuilder() function in xla::OutfeedReceiverImpl
467 StatusOr<XlaOp> OutfeedReceiver::AddOutfeedToBuilder( in AddOutfeedToBuilder() function in xla::OutfeedReceiver
474 return p_impl_->AddOutfeedToBuilder(builder, token, consumer_id, arrays); in AddOutfeedToBuilder()
Doutfeed_receiver.h65 StatusOr<XlaOp> AddOutfeedToBuilder(XlaBuilder* builder, XlaOp token,
Doutfeed_receiver_py.cc85 return outfeed_receiver_->AddOutfeedToBuilder(builder, token, consumer_id, in AddOutfeed()