Home
last modified time | relevance | path

Searched defs:recv_callbacks (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_executable.cc116 auto& recv_callbacks = in ExecuteInternal() local
215 auto& recv_callbacks = in ExecuteShardedOnLocalDevicesInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dhost_callback.h128 std::vector<std::vector<RecvCallback>> recv_callbacks; member
H A Dhost_callback_test.cc67 auto& recv_callbacks = states.recv_callbacks.emplace_back(); in TEST() local
H A Dhost_callback.cc105 std::vector<RecvCallback>& recv_callbacks) { in CreateHostCallbackStateAndAppendSendRecvCallbacks()
H A Dpjrt_client.h1092 absl::Span<const std::vector<RecvCallback>> recv_callbacks; member