Searched refs:ThenExecuteOnCallbackThread (Results 1 – 3 of 3) sorted by relevance
139 void ThenExecuteOnCallbackThread(se::Stream* stream,154 ThenExecuteOnCallbackThread(callback_stream_.get(), in ThenRelease()
99 void LocalDeviceState::ThenExecuteOnCallbackThread( in ThenExecuteOnCallbackThread() function in xla::LocalDeviceState
808 local_device->ThenExecuteOnCallbackThread( in BufferFromHostBuffer()1105 local_device_state->ThenExecuteOnCallbackThread( in Release()1799 device_state->ThenExecuteOnCallbackThread( in EnqueueExecution()1843 device_state->ThenExecuteOnCallbackThread( in MakeOutputBuffers()