Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Doutfeed_receiver_py.cc39 class OutfeedReceiverForPython { class
45 OutfeedReceiverForPython(CallbackToPython callback_python, in OutfeedReceiverForPython() function in xla::__anon99527ae70111::OutfeedReceiverForPython
63 OutfeedReceiverForPython(const OutfeedReceiverForPython&) = delete;
64 OutfeedReceiverForPython& operator=(const OutfeedReceiverForPython&) = delete;
66 ~OutfeedReceiverForPython() { in ~OutfeedReceiverForPython()
128 [](OutfeedReceiverForPython::CallbackToPython callback_to_python, in BuildOutfeedReceiverSubmodule()
131 -> std::unique_ptr<OutfeedReceiverForPython> { in BuildOutfeedReceiverSubmodule()
132 auto server = absl::make_unique<OutfeedReceiverForPython>( in BuildOutfeedReceiverSubmodule()
156 py::class_<OutfeedReceiverForPython> outfeed_receiver_class( in BuildOutfeedReceiverSubmodule()
160 "add_outfeed", &OutfeedReceiverForPython::AddOutfeed, py::arg("builder"), in BuildOutfeedReceiverSubmodule()