Home
last modified time | relevance | path

Searched refs:kROCm (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dplatform.cc31 case PlatformKind::kROCm: in PlatformKindString()
57 case PlatformKind::kROCm: in PlatformIsRunnable()
69 case PlatformKind::kROCm: in PlatformIsRunnableOnDevice()
Dplatform.h45 kROCm, enumerator
Dstream_executor_pimpl.cc157 platform_kind_ = PlatformKind::kROCm; in StreamExecutor()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc100 if (platform_kind == stream_executor::PlatformKind::kROCm) { in CheckCompatibilityWithServiceExecutableRunOptions()
Dgpu_conv_algorithm_picker.cc333 if (stream_exec_->platform_kind() == se::PlatformKind::kROCm) { in PickBestAlgorithm()