Searched refs:OutfeedReceiver (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | outfeed_receiver.h | 32 class OutfeedReceiver {
47 OutfeedReceiver(Callback callback, absl::Span<PjRtClient* const> clients,
50 OutfeedReceiver(const OutfeedReceiver&) = delete;
51 OutfeedReceiver& operator=(const OutfeedReceiver&) = delete;
55 ~OutfeedReceiver();
|
| D | outfeed_receiver_test.cc | 113 OutfeedReceiver::Callback callback = in TEST()
119 std::make_shared<OutfeedReceiver>(callback, clients, 128); in TEST()
146 OutfeedReceiver::Callback callback = in TEST()
152 std::make_shared<OutfeedReceiver>(callback, clients, 128); in TEST()
191 OutfeedReceiver::Callback callback = in TEST()
197 std::make_shared<OutfeedReceiver>(callback, clients, 128); in TEST()
234 OutfeedReceiver::Callback callback = in TEST()
240 std::make_shared<OutfeedReceiver>(callback, clients, 128); in TEST()
268 OutfeedReceiver::Callback callback = in TEST()
274 std::make_shared<OutfeedReceiver>(callback, clients, 128); in TEST()
[all …]
|
| D | outfeed_receiver.cc | 154 OutfeedReceiverImpl(OutfeedReceiver::Callback callback,
202 OutfeedReceiver::Callback callback_;
229 OutfeedReceiver::Callback callback, absl::Span<PjRtClient* const> clients, in OutfeedReceiverImpl()
456 OutfeedReceiver::OutfeedReceiver(Callback callback, in OutfeedReceiver() function in xla::OutfeedReceiver
463 OutfeedReceiver::~OutfeedReceiver() {} in ~OutfeedReceiver()
465 void OutfeedReceiver::Start() { p_impl_->Start(); } in Start()
467 StatusOr<XlaOp> OutfeedReceiver::AddOutfeedToBuilder( in AddOutfeedToBuilder()
|
| D | outfeed_receiver_py.cc | 50 OutfeedReceiver::Callback callback = in OutfeedReceiverForPython()
60 outfeed_receiver_ = absl::make_unique<OutfeedReceiver>( in OutfeedReceiverForPython()
118 std::unique_ptr<OutfeedReceiver> outfeed_receiver_;
|