Home
last modified time | relevance | path

Searched defs:compilation_device (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dtpu_rewrite_device_util.h76 std::string compilation_device; member
H A Dtpu_rewrite_device_util.cc457 std::string compilation_device = GetTPUCompilationDevice(system_devices[0]); in GetTPUCompilationAndExecutionDevices() local
H A Dtpu_rewrite_device_util_test.cc104 std::string compilation_device; in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_rewrite_pass.cc333 int num_cores_per_replica, llvm::StringRef compilation_device, in BuildCompileOp()
603 llvm::StringRef compilation_device, in BuildTPUCompileSucceededAssertOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Ddevice_util.cc69 const XlaOpRegistry::DeviceRegistration* compilation_device; in GetIdFor() local