Searched defs:definition_event (Results 1 – 8 of 8) sorted by relevance
367 std::shared_ptr<BufferSequencingEvent> definition_event = nullptr) { in AllocateDestinationBuffer()482 std::shared_ptr<BufferSequencingEvent> definition_event, in AddDestinationBufferSynchronization()905 std::shared_ptr<BufferSequencingEvent> definition_event) { in CreateUninitializedBuffer()1006 std::shared_ptr<BufferSequencingEvent> definition_event = in MakeCrossHostReceiveBuffers() local1054 std::shared_ptr<BufferSequencingEvent> definition_event = in MakeCrossHostReceiveBuffersForGather() local1700 std::shared_ptr<BufferSequencingEvent> definition_event, PjRtClient* client, in OutputBufferHelper()2009 std::shared_ptr<BufferSequencingEvent> definition_event, PjRtDevice* device, in MakeOutputBuffers()2108 auto definition_event = std::make_shared<BufferSequencingEvent>(); in ExecuteHelper() local
272 std::shared_ptr<BufferSequencingEvent> definition_event, in EnqueueCrossHostReceive()
539 tfrt::AsyncValueRef<CpuEvent> definition_event = in BufferFromHostLiteral() local1110 tfrt::RCReference<tfrt::IndirectAsyncValue> definition_event = in CopyToDevice() local
212 auto definition_event = std::make_shared<se::Event>(stream->parent()); in DoTpuExecute() local
399 const std::shared_ptr<se::Event>& definition_event) { in AllocateOutputTensors()719 auto definition_event = std::make_shared<se::Event>(stream->parent()); in DoWork() local
229 const std::shared_ptr<se::Event>& definition_event) { in UpdateOutputVariables()
324 bool use_multiple_streams, std::shared_ptr<se::Event> definition_event) { in GetOrCreateTensorForOutput()484 std::shared_ptr<se::Event> definition_event; in PopulateOutputs() local
263 auto definition_event = in TpuDeviceToDeviceCopy() local