/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
H A D | stream_executor_test_util.cc | 44 void CreateStreamDependency(const SP_Device* const device, SP_Stream dependent, in CreateStreamDependency() function 110 se->create_stream_dependency = CreateStreamDependency; in PopulateDefaultStreamExecutor()
|
H A D | stream_executor.cc | 476 bool CreateStreamDependency(Stream* dependent, Stream* other) override { in CreateStreamDependency() function in stream_executor::__anon702966b10111::CStreamExecutor
|
H A D | stream_executor_test.cc | 292 TEST_F(StreamExecutorTest, CreateStreamDependency) { in TEST_F() argument
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
H A D | executor.cc | 104 bool XlaInterpreterExecutor::CreateStreamDependency(Stream *dependent, in CreateStreamDependency() function in stream_executor::interpreter::XlaInterpreterExecutor
|
H A D | executor.h | 152 bool CreateStreamDependency(Stream *dependent, Stream *other) override;
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
H A D | host_gpu_executor.h | 118 bool CreateStreamDependency(Stream* dependent, Stream* other) override;
|
H A D | host_gpu_executor.cc | 193 bool HostExecutor::CreateStreamDependency(Stream* dependent, Stream* other) { in CreateStreamDependency() function in stream_executor::host::HostExecutor
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
H A D | tpu_executor.h | 84 bool CreateStreamDependency(Stream* dependent, Stream* other) override;
|
H A D | tpu_executor.cc | 102 bool TpuExecutor::CreateStreamDependency(Stream* dependent, Stream* other) { in CreateStreamDependency() function in tensorflow::tpu::TpuExecutor
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
H A D | stream_executor_internal.h | 254 virtual bool CreateStreamDependency(Stream* dependent, Stream* other) = 0;
|
H A D | stream_executor_pimpl.cc | 791 bool StreamExecutor::CreateStreamDependency(Stream* dependent, Stream* other) { in CreateStreamDependency() function in stream_executor::StreamExecutor 792 return implementation_->CreateStreamDependency(dependent, other); in CreateStreamDependency()
|
H A D | stream_executor_pimpl.h | 638 bool CreateStreamDependency(Stream* dependent, Stream* other);
|
H A D | stream.cc | 1033 CheckError(parent_->CreateStreamDependency(this, other)); in ThenWaitFor()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
H A D | gpu_executor.h | 214 bool CreateStreamDependency(Stream* dependent, Stream* other) override;
|
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
H A D | rocm_gpu_executor.cc | 635 bool GpuExecutor::CreateStreamDependency(Stream* dependent, Stream* other) { in CreateStreamDependency() function in stream_executor::gpu::GpuExecutor
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
H A D | cuda_gpu_executor.cc | 767 bool GpuExecutor::CreateStreamDependency(Stream* dependent, Stream* other) { in CreateStreamDependency() function in stream_executor::gpu::GpuExecutor
|