Searched refs:OutfeedReceiverForPython (Results 1 – 2 of 2) sorted by relevance
38 class OutfeedReceiverForPython { class44 OutfeedReceiverForPython(CallbackToPython callback_python, in OutfeedReceiverForPython() function in xla::__anon9362a2160111::OutfeedReceiverForPython62 OutfeedReceiverForPython(const OutfeedReceiverForPython&) = delete;63 OutfeedReceiverForPython& operator=(const OutfeedReceiverForPython&) = delete;65 ~OutfeedReceiverForPython() { in ~OutfeedReceiverForPython()127 [](OutfeedReceiverForPython::CallbackToPython callback_to_python, in BuildOutfeedReceiverSubmodule()130 -> std::unique_ptr<OutfeedReceiverForPython> { in BuildOutfeedReceiverSubmodule()131 auto server = std::make_unique<OutfeedReceiverForPython>( in BuildOutfeedReceiverSubmodule()155 py::class_<OutfeedReceiverForPython> outfeed_receiver_class( in BuildOutfeedReceiverSubmodule()159 "add_outfeed", &OutfeedReceiverForPython::AddOutfeed, py::arg("builder"), in BuildOutfeedReceiverSubmodule()
30 max_queue_size_bytes: int = ...) -> OutfeedReceiverForPython: ...32 class OutfeedReceiverForPython: