Searched refs:PlatformKind (Results 1 – 6 of 6) sorted by relevance
27 string PlatformKindString(PlatformKind kind) { in PlatformKindString()29 case PlatformKind::kCuda: in PlatformKindString()31 case PlatformKind::kROCm: in PlatformKindString()33 case PlatformKind::kOpenCL: in PlatformKindString()35 case PlatformKind::kHost: in PlatformKindString()37 case PlatformKind::kMock: in PlatformKindString()44 PlatformKind PlatformKindFromString(string kind) { in PlatformKindFromString()45 for (int i = 0; i < static_cast<int>(PlatformKind::kSize); ++i) { in PlatformKindFromString()46 if (kind == PlatformKindString(static_cast<PlatformKind>(i))) { in PlatformKindFromString()47 return static_cast<PlatformKind>(i); in PlatformKindFromString()[all …]
40 enum class PlatformKind { enum53 bool PlatformIsRunnable(PlatformKind kind);58 bool PlatformIsRunnableOnDevice(PlatformKind kind);61 string PlatformKindString(PlatformKind kind);65 PlatformKind PlatformKindFromString(string platform_string);68 void CheckPlatformKindIsValid(PlatformKind kind);
77 void MapPlatformKindToId(PlatformKind platform_kind,102 port::StatusOr<FactoryT> GetFactory(PlatformKind platform_kind,147 std::map<PlatformKind, Platform::Id> platform_id_by_kind_;
64 PlatformKind platform_kind, const PluginConfig &plugin_config) { in StreamExecutorImplementationFromPlatformKind()71 case PlatformKind::kCuda: in StreamExecutorImplementationFromPlatformKind()74 case PlatformKind::kROCm: in StreamExecutorImplementationFromPlatformKind()77 case PlatformKind::kOpenCL: in StreamExecutorImplementationFromPlatformKind()80 case PlatformKind::kHost: in StreamExecutorImplementationFromPlatformKind()157 StreamExecutor::StreamExecutor(PlatformKind 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()[all …]
63 void PluginRegistry::MapPlatformKindToId(PlatformKind platform_kind, in MapPlatformKindToId()232 PlatformKind platform_kind, PluginId plugin_id) { \
73 explicit StreamExecutor(PlatformKind kind,87 PlatformKind platform_kind() const { return platform_kind_; } in platform_kind()681 PlatformKind platform_kind_;