Searched refs:platform_kind (Results 1 – 5 of 5) sorted by relevance
66 void PluginRegistry::MapPlatformKindToId(PlatformKind platform_kind, in MapPlatformKindToId() argument68 platform_id_by_kind_[platform_kind] = platform_id; in MapPlatformKindToId()236 PlatformKind platform_kind, PluginId plugin_id) { \237 auto iter = platform_id_by_kind_.find(platform_kind); \241 static_cast<int>(platform_kind))); \
76 void MapPlatformKindToId(PlatformKind platform_kind,101 port::StatusOr<FactoryT> GetFactory(PlatformKind platform_kind,174 PlatformKind platform_kind, PluginId plugin_id)
87 PlatformKind platform_kind() const { return platform_kind_; } in platform_kind() function
98 stream_executor::PlatformKind platform_kind = in CheckCompatibilityWithServiceExecutableRunOptions() local99 main_stream->parent()->platform_kind(); in CheckCompatibilityWithServiceExecutableRunOptions()100 if (platform_kind == stream_executor::PlatformKind::kROCm) { in CheckCompatibilityWithServiceExecutableRunOptions()109 } else if (platform_kind == stream_executor::PlatformKind::kCuda) { in CheckCompatibilityWithServiceExecutableRunOptions()122 return InternalError("Unknown platform: %d", platform_kind); in CheckCompatibilityWithServiceExecutableRunOptions()273 if (executor->platform_kind() == se::PlatformKind::kCuda && in ResolveConstantGlobals()
333 if (stream_exec_->platform_kind() == se::PlatformKind::kROCm) { in PickBestAlgorithm()335 } else if (stream_exec_->platform_kind() == se::PlatformKind::kCuda) { in PickBestAlgorithm()