Searched refs:compilation_device (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.h | 64 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment()
66 : compilation_device(compilation_device), in TPUDeviceAssignment()
69 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment()
72 : compilation_device(compilation_device), in TPUDeviceAssignment()
76 std::string compilation_device; member
|
| D | tpu_rewrite_device_util.cc | 457 std::string compilation_device = GetTPUCompilationDevice(system_devices[0]); in GetTPUCompilationAndExecutionDevices() local
468 return TPUDeviceAssignment(compilation_device, in GetTPUCompilationAndExecutionDevices()
476 return TPUDeviceAssignment(compilation_device, in GetTPUCompilationAndExecutionDevices()
|
| D | tpu_rewrite_device_util_test.cc | 104 std::string compilation_device; in TEST_P() local
332 EXPECT_EQ(tpu_device_assignment.compilation_device, in TEST()
435 EXPECT_EQ(tpu_device_assignment.compilation_device, in TEST()
548 EXPECT_EQ(tpu_device_assignment.compilation_device, in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | device_util.cc | 69 const XlaOpRegistry::DeviceRegistration* compilation_device; in GetIdFor() local
71 &compilation_device)) { in GetIdFor()
72 compilation_device = nullptr; in GetIdFor()
74 id_to_compilation_device_.push_back(compilation_device); in GetIdFor()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 333 int num_cores_per_replica, llvm::StringRef compilation_device, in BuildCompileOp() argument
396 compilation_device); in BuildCompileOp()
603 llvm::StringRef compilation_device, in BuildTPUCompileSucceededAssertOp() argument
607 WrapOpInLaunch(builder, compile_op->getLoc(), assert_op, compilation_device); in BuildTPUCompileSucceededAssertOp()
771 tpu_device_assignment.compilation_device, in Rewrite()
815 compile_op, result_id, tpu_device_assignment.compilation_device, builder); in Rewrite()
|