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.cc162 platform_kind_(platform_kind), in StreamExecutor()
193 platform_kind_ = PlatformKind::kCuda; in StreamExecutor()
195 platform_kind_ = PlatformKind::kROCm; in StreamExecutor()
197 platform_kind_ = PlatformKind::kOpenCL; in StreamExecutor()
199 platform_kind_ = PlatformKind::kHost; in StreamExecutor()
201 platform_kind_ = PlatformKind::kInvalid; in StreamExecutor()
Dstream_executor_pimpl.h87 PlatformKind platform_kind() const { return platform_kind_; } in platform_kind()
681 PlatformKind platform_kind_; variable