Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/
H A Dstream_executor_test_util.cc44 void CreateStreamDependency(const SP_Device* const device, SP_Stream dependent, in CreateStreamDependency() function
110 se->create_stream_dependency = CreateStreamDependency; in PopulateDefaultStreamExecutor()
H A Dstream_executor.cc476 bool CreateStreamDependency(Stream* dependent, Stream* other) override { in CreateStreamDependency() function in stream_executor::__anon702966b10111::CStreamExecutor
H A Dstream_executor_test.cc292 TEST_F(StreamExecutorTest, CreateStreamDependency) { in TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutor.cc104 bool XlaInterpreterExecutor::CreateStreamDependency(Stream *dependent, in CreateStreamDependency() function in stream_executor::interpreter::XlaInterpreterExecutor
H A Dexecutor.h152 bool CreateStreamDependency(Stream *dependent, Stream *other) override;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
H A Dhost_gpu_executor.h118 bool CreateStreamDependency(Stream* dependent, Stream* other) override;
H A Dhost_gpu_executor.cc193 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 Dtpu_executor.h84 bool CreateStreamDependency(Stream* dependent, Stream* other) override;
H A Dtpu_executor.cc102 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 Dstream_executor_internal.h254 virtual bool CreateStreamDependency(Stream* dependent, Stream* other) = 0;
H A Dstream_executor_pimpl.cc791 bool StreamExecutor::CreateStreamDependency(Stream* dependent, Stream* other) { in CreateStreamDependency() function in stream_executor::StreamExecutor
792 return implementation_->CreateStreamDependency(dependent, other); in CreateStreamDependency()
H A Dstream_executor_pimpl.h638 bool CreateStreamDependency(Stream* dependent, Stream* other);
H A Dstream.cc1033 CheckError(parent_->CreateStreamDependency(this, other)); in ThenWaitFor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dgpu_executor.h214 bool CreateStreamDependency(Stream* dependent, Stream* other) override;
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc635 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 Dcuda_gpu_executor.cc767 bool GpuExecutor::CreateStreamDependency(Stream* dependent, Stream* other) { in CreateStreamDependency() function in stream_executor::gpu::GpuExecutor