Home
last modified time | relevance | path

Searched refs:SynchronizeAllActivity (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dlocal_device_state.cc67 Status status = SynchronizeAllActivity(); in ~LocalDeviceState()
73 Status LocalDeviceState::SynchronizeAllActivity() { in SynchronizeAllActivity() function in xla::LocalDeviceState
82 bool ok = compute_stream_->parent()->SynchronizeAllActivity(); in SynchronizeAllActivity()
Dlocal_device_state.h164 Status SynchronizeAllActivity();
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h91 bool SynchronizeAllActivity() override { return true; } in SynchronizeAllActivity() function
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h106 bool SynchronizeAllActivity() override { return true; } in SynchronizeAllActivity() function
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.h133 bool SynchronizeAllActivity() override;
Dtpu_executor.cc59 bool TpuExecutor::SynchronizeAllActivity() { in SynchronizeAllActivity() function in tensorflow::tpu::TpuExecutor
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h218 virtual bool SynchronizeAllActivity() = 0;
Dstream_executor_pimpl.cc579 bool StreamExecutor::SynchronizeAllActivity() { in SynchronizeAllActivity() function in stream_executor::StreamExecutor
582 bool ok = implementation_->SynchronizeAllActivity(); in SynchronizeAllActivity()
Dstream_executor_pimpl.h233 bool SynchronizeAllActivity() SE_MUST_USE_RESULT;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_algorithm_picker.cc63 if (!stream->parent()->SynchronizeAllActivity()) { in DoUncachedGemmAutotune()
Dgpu_executable.cc89 CHECK(pair.first->SynchronizeAllActivity()); in ~GpuExecutable()
Dgpu_conv_algorithm_picker.cc312 if (!stream_exec_->SynchronizeAllActivity()) { in PickBestAlgorithm()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h120 bool SynchronizeAllActivity() override;
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc361 if (!dev_info->stream->parent()->SynchronizeAllActivity() || in SyncAll()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.cc65 bool ok = executor->SynchronizeAllActivity(); in ~XlaCompilationCache()
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test.cc876 TEST_F(StreamExecutorTest, SynchronizeAllActivity) { in TEST_F() argument
886 ASSERT_TRUE(executor->SynchronizeAllActivity()); in TEST_F()
Dstream_executor.cc332 bool SynchronizeAllActivity() override { in SynchronizeAllActivity() function in stream_executor::__anon00950a620111::CStreamExecutor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc454 bool GpuExecutor::SynchronizeAllActivity() { in SynchronizeAllActivity() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc546 bool GpuExecutor::SynchronizeAllActivity() { in SynchronizeAllActivity() function in stream_executor::gpu::GpuExecutor