Home
last modified time | relevance | path

Searched defs:definition_event (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_tfrt_cpu_device_buffer_test.cc33 auto definition_event = tfrt::MakeConstructedAsyncValueRef<CpuEvent>(); in TEST() local
103 auto definition_event = tfrt::MakeConstructedAsyncValueRef<CpuEvent>(); in TEST() local
Dtracked_tfrt_cpu_device_buffer.h117 const tfrt::AsyncValueRef<CpuEvent>& definition_event() const { in definition_event() function
Dpjrt_stream_executor_client.cc377 std::shared_ptr<BufferSequencingEvent> definition_event = nullptr) { in AllocateDestinationBuffer()
492 std::shared_ptr<BufferSequencingEvent> definition_event, in AddDestinationBufferSynchronization()
915 std::shared_ptr<BufferSequencingEvent> definition_event) { in CreateUninitializedBuffer()
1011 std::shared_ptr<BufferSequencingEvent> definition_event = in MakeCrossHostReceiveBuffers() local
1053 std::shared_ptr<BufferSequencingEvent> definition_event = in MakeCrossHostReceiveBuffersForGather() local
1746 std::shared_ptr<BufferSequencingEvent> definition_event, PjRtClient* client, in OutputBufferHelper()
2055 std::shared_ptr<BufferSequencingEvent> definition_event, PjRtDevice* device, in MakeOutputBuffers()
2153 auto definition_event = std::make_shared<BufferSequencingEvent>(); in ExecuteHelper() local
Dtracked_tfrt_cpu_device_buffer.cc85 tfrt::AsyncValueRef<CpuEvent> definition_event, in TrackedTfrtCpuDeviceBuffer()
Dtfrt_cpu_pjrt_client.cc601 tfrt::AsyncValueRef<CpuEvent> definition_event = in BufferFromHostLiteral() local
1108 tfrt::AsyncValueRef<CpuEvent> definition_event; in GetReadyFuture() local
1404 const auto& definition_event = tracked_buffer->definition_event(); in ExecuteHelper() local
Dpjrt_stream_executor_client.h297 std::shared_ptr<BufferSequencingEvent> definition_event, in EnqueueCrossHostReceive()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_reshard_variables_op.cc211 auto definition_event = std::make_shared<se::Event>(stream->parent()); in DoTpuExecute() local
Dtpu_execute_op.cc398 const std::shared_ptr<se::Event>& definition_event) { in AllocateOutputTensors()
718 auto definition_event = std::make_shared<se::Event>(stream->parent()); in DoWork() local
Dtpu_reshard_variables_op_util.cc229 const std::shared_ptr<se::Event>& definition_event) { in UpdateOutputVariables()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc328 bool use_multiple_streams, std::shared_ptr<se::Event> definition_event) { in GetOrCreateTensorForOutput()
488 std::shared_ptr<se::Event> definition_event; in PopulateOutputs() local
Dxla_tpu_device.cc271 auto definition_event = in TpuDeviceToDeviceCopy() local