Searched refs:definition_event (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 367 std::shared_ptr<BufferSequencingEvent> definition_event = nullptr) { in AllocateDestinationBuffer() argument
409 if (definition_event) { in AllocateDestinationBuffer()
410 definition_events.emplace_back(definition_event); in AllocateDestinationBuffer()
415 if (definition_event) { in AllocateDestinationBuffer()
416 definition_events.emplace_back(definition_event); in AllocateDestinationBuffer()
482 std::shared_ptr<BufferSequencingEvent> definition_event, in AddDestinationBufferSynchronization() argument
490 definition_event->SetSequencingEvent(event_or.ConsumeValueOrDie(), in AddDestinationBufferSynchronization()
499 definition_event, copy_stream, in AddDestinationBufferSynchronization()
836 std::shared_ptr<BufferSequencingEvent> definition_event) { in CreateUninitializedBuffer() argument
854 definition_event)); in CreateUninitializedBuffer()
[all …]
|
| D | pjrt_stream_executor_client.h | 185 std::shared_ptr<BufferSequencingEvent> definition_event);
238 std::shared_ptr<BufferSequencingEvent> definition_event, in EnqueueCrossHostReceive() argument
698 std::shared_ptr<BufferSequencingEvent> definition_event,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 333 std::shared_ptr<se::Event> definition_event) { in PopulateXlaTensor() argument
338 xla_tensor->ResetDefinitionEvent(definition_event, stream); in PopulateXlaTensor()
452 std::shared_ptr<se::Event> definition_event; in PopulateOutputs() local
454 definition_event = std::make_shared<se::Event>(stream->parent()); in PopulateOutputs()
455 if (!definition_event->Init()) { in PopulateOutputs()
458 stream->ThenRecordEvent(definition_event.get()); in PopulateOutputs()
514 use_multiple_streams_, definition_event); in PopulateOutputs()
561 use_multiple_streams_, definition_event); in PopulateOutputs()
|
| D | xla_tpu_device.cc | 263 auto definition_event = in TpuDeviceToDeviceCopy() local
265 TF_RET_CHECK(definition_event->Init()) << "Event failed to initialize!"; in TpuDeviceToDeviceCopy()
266 dst_device_to_device_stream->ThenRecordEvent(definition_event.get()); in TpuDeviceToDeviceCopy()
267 xla_output->ResetDefinitionEvent(std::move(definition_event), in TpuDeviceToDeviceCopy()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 399 const std::shared_ptr<se::Event>& definition_event) { in AllocateOutputTensors() argument
481 xla_tensor->ResetDefinitionEvent(definition_event, stream); in AllocateOutputTensors()
723 auto definition_event = std::make_shared<se::Event>(stream->parent()); in DoWork() local
724 TF_RET_CHECK(definition_event->Init()) in DoWork()
762 stream->ThenRecordEvent(definition_event.get()); in DoWork()
769 input_buffers.get(), definition_event)); in DoWork()
|