Home
last modified time | relevance | path

Searched refs:platform_kind (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.cc63 void PluginRegistry::MapPlatformKindToId(PlatformKind platform_kind, in MapPlatformKindToId() argument
65 platform_id_by_kind_[platform_kind] = platform_id; in MapPlatformKindToId()
232 PlatformKind platform_kind, PluginId plugin_id) { \
233 auto iter = platform_id_by_kind_.find(platform_kind); \
237 static_cast<int>(platform_kind))); \
Dplugin_registry.h77 void MapPlatformKindToId(PlatformKind platform_kind,
102 port::StatusOr<FactoryT> GetFactory(PlatformKind platform_kind,
Dstream_executor_pimpl.cc64 PlatformKind platform_kind, const PluginConfig &plugin_config) { in StreamExecutorImplementationFromPlatformKind() argument
70 switch (platform_kind) { in StreamExecutorImplementationFromPlatformKind()
89 << PlatformKindString(platform_kind); in StreamExecutorImplementationFromPlatformKind()
157 StreamExecutor::StreamExecutor(PlatformKind platform_kind, in StreamExecutor() argument
161 platform_kind, plugin_config)), in StreamExecutor()
162 platform_kind_(platform_kind), in StreamExecutor()
168 CheckPlatformKindIsValid(platform_kind); in StreamExecutor()
Dstream_executor_pimpl.h87 PlatformKind platform_kind() const { return platform_kind_; } in platform_kind() function