Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_executable.cc114 auto& send_callbacks = in ExecuteInternal() local
213 auto& send_callbacks = in ExecuteShardedOnLocalDevicesInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dhost_callback.h127 std::vector<std::vector<SendCallback>> send_callbacks; member
H A Dhost_callback_test.cc66 auto& send_callbacks = states.send_callbacks.emplace_back(); in TEST() local
H A Dhost_callback.cc104 std::vector<SendCallback>& send_callbacks, in CreateHostCallbackStateAndAppendSendRecvCallbacks()
H A Dpjrt_client.h1091 absl::Span<const std::vector<SendCallback>> send_callbacks; member