Home
last modified time | relevance | path

Searched refs:platform_kind_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc155 platform_kind_ = PlatformKind::kCuda; in StreamExecutor()
157 platform_kind_ = PlatformKind::kROCm; in StreamExecutor()
159 platform_kind_ = PlatformKind::kOpenCL; in StreamExecutor()
161 platform_kind_ = PlatformKind::kHost; in StreamExecutor()
163 platform_kind_ = PlatformKind::kInvalid; in StreamExecutor()
Dstream_executor_pimpl.h87 PlatformKind platform_kind() const { return platform_kind_; } in platform_kind()
707 PlatformKind platform_kind_; variable