Home
last modified time | relevance | path

Searched refs:IsTPUDevice (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_device_propagation.cc146 !tensorflow::IsTPUDevice(arg_device_attr.getValue())) in PropagateDevicesFromArguments()
177 if (has_device && tensorflow::IsTPUDevice(device_attr.getValue())) { in PropagateDevicesInGraph()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.h251 bool IsTPUDevice(llvm::StringRef device);
Dtpu_rewrite_device_util_test.cc810 EXPECT_TRUE(IsTPUDevice("/job:localhost/replica:0/task:0/device:TPU:0")); in TEST()
811 EXPECT_FALSE(IsTPUDevice("/job:localhost/replica:0/task:0/device:CPU:0")); in TEST()
812 EXPECT_FALSE(IsTPUDevice("INVALID_DEVICE")); in TEST()
Dtpu_rewrite_device_util.cc544 bool IsTPUDevice(llvm::StringRef device) { in IsTPUDevice() function