Searched defs:definition_event (Results 1 – 5 of 5) sorted by relevance
367 std::shared_ptr<BufferSequencingEvent> definition_event = nullptr) { in AllocateDestinationBuffer()482 std::shared_ptr<BufferSequencingEvent> definition_event, in AddDestinationBufferSynchronization()836 std::shared_ptr<BufferSequencingEvent> definition_event) { in CreateUninitializedBuffer()937 std::shared_ptr<BufferSequencingEvent> definition_event = in MakeCrossHostReceiveBuffers() local1552 std::shared_ptr<BufferSequencingEvent> definition_event, PjRtClient* client, in OutputBufferHelper()1825 std::shared_ptr<BufferSequencingEvent> definition_event, in MakeOutputBuffers()1917 auto definition_event = std::make_shared<BufferSequencingEvent>(); in ExecuteHelper() local
238 std::shared_ptr<BufferSequencingEvent> definition_event, in EnqueueCrossHostReceive()
399 const std::shared_ptr<se::Event>& definition_event) { in AllocateOutputTensors()723 auto definition_event = std::make_shared<se::Event>(stream->parent()); in DoWork() local
333 std::shared_ptr<se::Event> definition_event) { in PopulateXlaTensor()452 std::shared_ptr<se::Event> definition_event; in PopulateOutputs() local
263 auto definition_event = in TpuDeviceToDeviceCopy() local