Searched defs:host_callback (Results 1 – 7 of 7) sorted by relevance
86 HostCallbackContext(HostCallback host_callback, PjRtClient* client) in HostCallbackContext() argument91 HostCallback host_callback, in HostCallbackContext()111 const HostCallback& host_callback() const { return host_callback_; } in host_callback() function
52 HostCallback host_callback; in TEST() local
102 HostCallback host_callback, in CreateHostCallbackStateAndAppendSendRecvCallbacks()
119 for (const py::capsule& host_callback : host_callbacks_) { in ExecuteInternal() local218 for (const py::capsule& host_callback : host_callbacks_) { in ExecuteShardedOnLocalDevicesInternal() local
556 auto* host_callback = new HostCallback; in MakePythonCallbackUsingHostSendAndRecv() local
179 static void host_callback(void *arg, int status, int timeouts, in host_callback() function
413 TF_Bool (*host_callback)(const SP_Device* device, SP_Stream stream, member