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 | 62 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment()
64 : compilation_device(compilation_device), in TPUDeviceAssignment()
67 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment()
70 : compilation_device(compilation_device), in TPUDeviceAssignment()
74 std::string compilation_device; member
|
| D | tpu_rewrite_device_util.cc | 461 std::string compilation_device = GetTPUCompilationDevice(system_devices[0]); in GetTPUCompilationAndExecutionDevices() local
472 return TPUDeviceAssignment(compilation_device, in GetTPUCompilationAndExecutionDevices()
480 return TPUDeviceAssignment(compilation_device, in GetTPUCompilationAndExecutionDevices()
|
| D | tpu_rewrite_device_util_test.cc | 103 std::string compilation_device; in TEST_P() local
331 EXPECT_EQ(tpu_device_assignment.compilation_device, in TEST()
434 EXPECT_EQ(tpu_device_assignment.compilation_device, in TEST()
547 EXPECT_EQ(tpu_device_assignment.compilation_device, in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | device_util.cc | 70 const XlaOpRegistry::DeviceRegistration* compilation_device; in GetIdFor() local
72 &compilation_device)) { in GetIdFor()
73 compilation_device = nullptr; in GetIdFor()
75 id_to_compilation_device_.push_back(compilation_device); in GetIdFor()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 351 int num_cores_per_replica, llvm::StringRef compilation_device, in BuildCompileOp() argument
406 compilation_device); in BuildCompileOp()
559 llvm::StringRef compilation_device, in BuildTPUCompileSucceededAssertOp() argument
563 WrapOpInLaunch(builder, compile_op->getLoc(), assert_op, compilation_device); in BuildTPUCompileSucceededAssertOp()
631 tpu_device_assignment.compilation_device, in Rewrite()
657 compile_op, tpu_device_assignment.compilation_device, builder); in Rewrite()
|