Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dplatform.cc29 case PlatformKind::kCuda: in PlatformKindString()
56 case PlatformKind::kCuda: in PlatformIsRunnable()
68 case PlatformKind::kCuda: in PlatformIsRunnableOnDevice()
78 CHECK(static_cast<int>(PlatformKind::kCuda) <= static_cast<int>(kind) && in CheckPlatformKindIsValid()
Dplatform.h44 kCuda, enumerator
Dstream_executor_pimpl.cc155 platform_kind_ = PlatformKind::kCuda; in StreamExecutor()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc109 } else if (platform_kind == stream_executor::PlatformKind::kCuda) { in CheckCompatibilityWithServiceExecutableRunOptions()
273 if (executor->platform_kind() == se::PlatformKind::kCuda && in ResolveConstantGlobals()
Dgpu_conv_algorithm_picker.cc335 } else if (stream_exec_->platform_kind() == se::PlatformKind::kCuda) { in PickBestAlgorithm()