Searched refs:kDeviceAssignmentAttr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 47 const char* const kDeviceAssignmentAttr = "device_assignment"; variable
362 "length of '", kDeviceAssignmentAttr, in GetGeneralTPUExecutionDeviceAssignment()
394 return DeviceCoordinateErrorMsg(kDeviceAssignmentAttr, x, y, z, core, in GetGeneralTPUExecutionDeviceAssignment()
402 "no TPU device found for '", kDeviceAssignmentAttr, in GetGeneralTPUExecutionDeviceAssignment()
407 return DuplicateCoordinateErrorMsg(kDeviceAssignmentAttr, x, y, z, in GetGeneralTPUExecutionDeviceAssignment()
439 llvm::formatv(kBadIntArrayElementMsg, kDeviceAssignmentAttr, in GetDeviceCoordinates()
465 return errors::InvalidArgument("'", kDeviceAssignmentAttr, in GetTPUCompilationAndExecutionDevices()
514 tensorflow::kDeviceAssignmentAttr); in GetHostDeviceOutsideComputation()
517 tensorflow::kDeviceAssignmentAttr) in GetHostDeviceOutsideComputation()
|
| D | tpu_rewrite_device_util.h | 40 extern const char* const kDeviceAssignmentAttr;
|
| D | tpu_rewrite_device_util_test.cc | 656 cluster->setAttr(kDeviceAssignmentAttr, builder.getArrayAttr({})); in TEST()
676 cluster->setAttr(kDeviceAssignmentAttr, builder.getArrayAttr({})); in TEST()
717 cluster->setAttr(kDeviceAssignmentAttr, in TEST()
744 cluster->setAttr(kDeviceAssignmentAttr, builder.getArrayAttr({})); in TEST()
799 cluster->setAttr(kDeviceAssignmentAttr, builder.getArrayAttr({})); in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 591 tensorflow::kDeviceAssignmentAttr); in Rewrite()
595 tensorflow::kDeviceAssignmentAttr) in Rewrite()
|