/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
D | local_device_state.cc | 67 Status status = SynchronizeAllActivity(); in ~LocalDeviceState() 73 Status LocalDeviceState::SynchronizeAllActivity() { in SynchronizeAllActivity() function in xla::LocalDeviceState 82 bool ok = compute_stream_->parent()->SynchronizeAllActivity(); in SynchronizeAllActivity()
|
D | local_device_state.h | 164 Status SynchronizeAllActivity();
|
/external/tensorflow/tensorflow/stream_executor/host/ |
D | host_gpu_executor.h | 91 bool SynchronizeAllActivity() override { return true; } in SynchronizeAllActivity() function
|
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
D | executor.h | 106 bool SynchronizeAllActivity() override { return true; } in SynchronizeAllActivity() function
|
/external/tensorflow/tensorflow/stream_executor/tpu/ |
D | tpu_executor.h | 133 bool SynchronizeAllActivity() override;
|
D | tpu_executor.cc | 59 bool TpuExecutor::SynchronizeAllActivity() { in SynchronizeAllActivity() function in tensorflow::tpu::TpuExecutor
|
/external/tensorflow/tensorflow/stream_executor/ |
D | stream_executor_internal.h | 218 virtual bool SynchronizeAllActivity() = 0;
|
D | stream_executor_pimpl.cc | 579 bool StreamExecutor::SynchronizeAllActivity() { in SynchronizeAllActivity() function in stream_executor::StreamExecutor 582 bool ok = implementation_->SynchronizeAllActivity(); in SynchronizeAllActivity()
|
D | stream_executor_pimpl.h | 233 bool SynchronizeAllActivity() SE_MUST_USE_RESULT;
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gemm_algorithm_picker.cc | 63 if (!stream->parent()->SynchronizeAllActivity()) { in DoUncachedGemmAutotune()
|
D | gpu_executable.cc | 89 CHECK(pair.first->SynchronizeAllActivity()); in ~GpuExecutable()
|
D | gpu_conv_algorithm_picker.cc | 312 if (!stream_exec_->SynchronizeAllActivity()) { in PickBestAlgorithm()
|
/external/tensorflow/tensorflow/stream_executor/gpu/ |
D | gpu_executor.h | 120 bool SynchronizeAllActivity() override;
|
/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
D | gpu_util.cc | 361 if (!dev_info->stream->parent()->SynchronizeAllActivity() || in SyncAll()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_compilation_cache.cc | 65 bool ok = executor->SynchronizeAllActivity(); in ~XlaCompilationCache()
|
/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
D | stream_executor_test.cc | 876 TEST_F(StreamExecutorTest, SynchronizeAllActivity) { in TEST_F() argument 886 ASSERT_TRUE(executor->SynchronizeAllActivity()); in TEST_F()
|
D | stream_executor.cc | 332 bool SynchronizeAllActivity() override { in SynchronizeAllActivity() function in stream_executor::__anon00950a620111::CStreamExecutor
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_gpu_executor.cc | 454 bool GpuExecutor::SynchronizeAllActivity() { in SynchronizeAllActivity() function in stream_executor::gpu::GpuExecutor
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_gpu_executor.cc | 546 bool GpuExecutor::SynchronizeAllActivity() { in SynchronizeAllActivity() function in stream_executor::gpu::GpuExecutor
|