Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_executable.cc120 contexts.push_back(CreateHostCallbackStateAndAppendSendRecvCallbacks( in ExecuteInternal()
219 contexts.push_back(CreateHostCallbackStateAndAppendSendRecvCallbacks( in ExecuteShardedOnLocalDevicesInternal()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dhost_callback_test.cc71 auto context = CreateHostCallbackStateAndAppendSendRecvCallbacks( in TEST()
H A Dhost_callback.h133 CreateHostCallbackStateAndAppendSendRecvCallbacks(
H A Dhost_callback.cc101 CreateHostCallbackStateAndAppendSendRecvCallbacks( in CreateHostCallbackStateAndAppendSendRecvCallbacks() function