Home
last modified time | relevance | path

Searched refs:definition_events (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer.cc100 definition_events) { in FromScopedShapedBuffer()
116 absl::Span<se::DeviceMemoryBase>(buffers), definition_events, in FromScopedShapedBuffer()
166 absl::Span<const std::shared_ptr<BufferSequencingEvent>> definition_events, in TrackedDeviceBuffer() argument
171 definition_events_(std::make_move_iterator(definition_events.begin()), in TrackedDeviceBuffer()
172 std::make_move_iterator(definition_events.end())), in TrackedDeviceBuffer()
222 for (const auto& e : buffer.definition_events()) { in GetDeviceBufferEvents()
Dtracked_device_buffer.h138 definition_events);
174 absl::Span<const std::shared_ptr<BufferSequencingEvent>> definition_events() in definition_events() function
210 definition_events,
Dpjrt_stream_executor_client.cc393 definition_events; in AllocateDestinationBuffer() local
401 definition_events.emplace_back(std::make_shared<BufferSequencingEvent>()); in AllocateDestinationBuffer()
405 definition_events.back()->SetSequencingEvent( in AllocateDestinationBuffer()
410 definition_events.emplace_back(definition_event); in AllocateDestinationBuffer()
416 definition_events.emplace_back(definition_event); in AllocateDestinationBuffer()
418 definition_events.emplace_back(std::make_shared<BufferSequencingEvent>()); in AllocateDestinationBuffer()
441 definition_events.emplace_back(std::make_shared<BufferSequencingEvent>()); in AllocateDestinationBuffer()
449 definition_events.back()->SetSequencingEvent(event_or.ConsumeValueOrDie(), in AllocateDestinationBuffer()
454 definition_events); in AllocateDestinationBuffer()
469 definition_events.back(), tuple_table_stream, in AllocateDestinationBuffer()
[all …]