Home
last modified time | relevance | path

Searched refs:GetRocmComputeCapability (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
H A Dtf_framework_c_interface.cc273 ctx->op_device_context()->stream()->GetRocmComputeCapability(); in _mlir_ciface_tf_jit_compile()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc291 auto cc = main_stream->GetRocmComputeCapability(); in CheckCompatibilityWithServiceExecutableRunOptions()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream.h1469 RocmComputeCapability GetRocmComputeCapability() const { in GetRocmComputeCapability() function