Home
last modified time | relevance | path

Searched refs:AllocateEvent (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Devent_pool.cc34 StatusOr<EventPool::Handle> EventPool::AllocateEvent( in AllocateEvent() function in xla::EventPool
62 AllocateEvent(stream->parent())); in ThenAllocateAndRecordEvent()
Devent_pool.h82 StatusOr<Handle> AllocateEvent(se::StreamExecutor* executor);
Dpjrt_stream_executor_client.cc1238 local_device->event_pool().AllocateEvent(stream->parent()); in ToLiteral()
/external/tensorflow/tensorflow/stream_executor/
Devent.cc40 auto status = stream_exec_->AllocateEvent(this); in Init()
Dstream_executor_internal.h249 virtual port::Status AllocateEvent(Event *event) = 0;
Dstream_executor_pimpl.cc742 port::Status StreamExecutor::AllocateEvent(Event *event) { in AllocateEvent() function in stream_executor::StreamExecutor
743 return implementation_->AllocateEvent(event); in AllocateEvent()
Dstream_executor_pimpl.h597 port::Status AllocateEvent(Event *event);
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h110 port::Status AllocateEvent(Event *event) override;
Dhost_gpu_executor.cc229 port::Status HostExecutor::AllocateEvent(Event * /*event*/) { in AllocateEvent() function in stream_executor::host::HostExecutor
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h131 port::Status AllocateEvent(Event *event) override { in AllocateEvent() function
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.h71 Status AllocateEvent(Event* event) override;
Dtpu_executor.cc108 Status TpuExecutor::AllocateEvent(Event* event) { return Status::OK(); } in AllocateEvent() function in tensorflow::tpu::TpuExecutor
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h173 port::Status AllocateEvent(Event* event) override;
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc579 port::Status GpuExecutor::AllocateEvent(Event* event) { in AllocateEvent() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc447 port::Status AllocateEvent(Event* event) override { in AllocateEvent() function in stream_executor::__anon00950a620111::CStreamExecutor
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc671 port::Status GpuExecutor::AllocateEvent(Event* event) { in AllocateEvent() function in stream_executor::gpu::GpuExecutor