Home
last modified time | relevance | path

Searched refs:ThenRecordEvent (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Devent_pool.cc53 void EventPool::ThenRecordEvent(se::Stream* stream, EventPool::Handle& handle) { in ThenRecordEvent() function in xla::EventPool
55 stream->ThenRecordEvent(handle.event_.get()); in ThenRecordEvent()
63 ThenRecordEvent(stream, handle); in ThenAllocateAndRecordEvent()
Devent_pool.h83 void ThenRecordEvent(se::Stream* stream, EventPool::Handle& handle);
Dpjrt_stream_executor_client.cc1251 local_device->event_pool().ThenRecordEvent(stream, event_or.ValueOrDie()); in ToLiteral()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr.cc179 stream->ThenRecordEvent(e); in QueueInUse()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr.cc180 stream->ThenRecordEvent(e); in QueueInUse()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc175 host_to_device_stream_->ThenRecordEvent(event.get()); in CopyCPUTensorToDevice()
Dxla_tpu_device.cc266 dst_device_to_device_stream->ThenRecordEvent(definition_event.get()); in TpuDeviceToDeviceCopy()
Dxla_launch_util.cc458 stream->ThenRecordEvent(definition_event.get()); in PopulateOutputs()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc203 stream->ThenRecordEvent(finish_event.get()); in ExecuteThunks()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc762 stream->ThenRecordEvent(definition_event.get()); in DoWork()
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test.cc593 stream.ThenRecordEvent(&event); in TEST_F()
/external/tensorflow/tensorflow/stream_executor/
Dstream.h239 Stream &ThenRecordEvent(Event *event);
Dstream.cc329 Stream &Stream::ThenRecordEvent(Event *event) { in ThenRecordEvent() function in stream_executor::Stream