Home
last modified time | relevance | path

Searched refs:LockGpu (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.h61 tensorflow::mutex_lock LockGpu(const se::StreamExecutor* stream_exec);
Dgemm_algorithm_picker.cc238 tensorflow::mutex_lock gpu_lock = LockGpu(stream->parent()); in DoGemmAutotune()
Dstream_executor_util.cc177 tensorflow::mutex_lock LockGpu(const se::StreamExecutor* stream_exec) { in LockGpu() function
Dgpu_conv_algorithm_picker.cc293 tensorflow::mutex_lock lock = LockGpu(stream_exec_); in PickBestAlgorithm()