Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/gpu/
H A Dgpu_utils.cc42 auto platform_kind = device->executor()->platform_kind(); in GetTfrtGpuPlatform() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dplugin_registry.cc66 void PluginRegistry::MapPlatformKindToId(PlatformKind platform_kind, in MapPlatformKindToId()
H A Dstream_executor_pimpl.h89 PlatformKind platform_kind() const { return platform_kind_; } in platform_kind() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc288 stream_executor::PlatformKind platform_kind = in CheckCompatibilityWithServiceExecutableRunOptions() local