Searched refs:ReffedStatusCallback (Results 1 – 7 of 7) sorted by relevance
36 auto* cb = new ReffedStatusCallback(std::move(done)); in TEST()50 auto* cb = new ReffedStatusCallback(std::move(done)); in TEST()66 auto* cb = new ReffedStatusCallback(std::move(done)); in TEST()90 auto* cb = new ReffedStatusCallback(std::move(done)); in TEST()
29 class ReffedStatusCallback : public core::RefCounted {31 explicit ReffedStatusCallback(StatusCallback done) in ReffedStatusCallback() function52 ~ReffedStatusCallback() { done_(status_); } in ~ReffedStatusCallback()
94 ReffedStatusCallback* reffed_status_callback_;114 reffed_status_callback_ = new ReffedStatusCallback([this](const Status& s) { in CallContainer()
56 auto* status_cb = new ReffedStatusCallback(std::move(done)); in CopyHostToDevice()121 auto* status_cb = new ReffedStatusCallback(std::move(done)); in CopyDeviceToHost()190 auto* status_cb = new ReffedStatusCallback(std::move(done)); in CopyDeviceToDevice()
91 auto status_cb = new ReffedStatusCallback(std::move(done)); in RecvOutputsFromRendezvousAsync()
766 auto* refcounted_done = new ReffedStatusCallback(std::move(done)); in RunMultiDevice()
92 auto* refcounted = new ReffedStatusCallback(std::move(callback)); in ComputeAsync()