/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
D | event_pool.cc | 34 StatusOr<EventPool::Handle> EventPool::AllocateEvent( in AllocateEvent() function in xla::EventPool 62 AllocateEvent(stream->parent())); in ThenAllocateAndRecordEvent()
|
D | event_pool.h | 82 StatusOr<Handle> AllocateEvent(se::StreamExecutor* executor);
|
D | pjrt_stream_executor_client.cc | 1238 local_device->event_pool().AllocateEvent(stream->parent()); in ToLiteral()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | event.cc | 40 auto status = stream_exec_->AllocateEvent(this); in Init()
|
D | stream_executor_internal.h | 249 virtual port::Status AllocateEvent(Event *event) = 0;
|
D | stream_executor_pimpl.cc | 742 port::Status StreamExecutor::AllocateEvent(Event *event) { in AllocateEvent() function in stream_executor::StreamExecutor 743 return implementation_->AllocateEvent(event); in AllocateEvent()
|
D | stream_executor_pimpl.h | 597 port::Status AllocateEvent(Event *event);
|
/external/tensorflow/tensorflow/stream_executor/host/ |
D | host_gpu_executor.h | 110 port::Status AllocateEvent(Event *event) override;
|
D | host_gpu_executor.cc | 229 port::Status HostExecutor::AllocateEvent(Event * /*event*/) { in AllocateEvent() function in stream_executor::host::HostExecutor
|
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
D | executor.h | 131 port::Status AllocateEvent(Event *event) override { in AllocateEvent() function
|
/external/tensorflow/tensorflow/stream_executor/tpu/ |
D | tpu_executor.h | 71 Status AllocateEvent(Event* event) override;
|
D | tpu_executor.cc | 108 Status TpuExecutor::AllocateEvent(Event* event) { return Status::OK(); } in AllocateEvent() function in tensorflow::tpu::TpuExecutor
|
/external/tensorflow/tensorflow/stream_executor/gpu/ |
D | gpu_executor.h | 173 port::Status AllocateEvent(Event* event) override;
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_gpu_executor.cc | 579 port::Status GpuExecutor::AllocateEvent(Event* event) { in AllocateEvent() function in stream_executor::gpu::GpuExecutor
|
/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
D | stream_executor.cc | 447 port::Status AllocateEvent(Event* event) override { in AllocateEvent() function in stream_executor::__anon00950a620111::CStreamExecutor
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_gpu_executor.cc | 671 port::Status GpuExecutor::AllocateEvent(Event* event) { in AllocateEvent() function in stream_executor::gpu::GpuExecutor
|