Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.h74 std::string compilation_device; member
Dtpu_rewrite_device_util_test.cc103 std::string compilation_device; in TEST_P() local
Dtpu_rewrite_device_util.cc461 std::string compilation_device = GetTPUCompilationDevice(system_devices[0]); in GetTPUCompilationAndExecutionDevices() local
/external/tensorflow/tensorflow/compiler/jit/
Ddevice_util.cc70 const XlaOpRegistry::DeviceRegistration* compilation_device; in GetIdFor() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc351 int num_cores_per_replica, llvm::StringRef compilation_device, in BuildCompileOp()
559 llvm::StringRef compilation_device, in BuildTPUCompileSucceededAssertOp()