| /external/tensorflow/tensorflow/core/util/ |
| D | device_name_utils_test.cc | 365 TEST(DeviceNameUtilsTest, SplitDeviceName) { in TEST() argument
368 EXPECT_TRUE(DeviceNameUtils::SplitDeviceName( in TEST()
372 EXPECT_TRUE(DeviceNameUtils::SplitDeviceName( in TEST()
377 DeviceNameUtils::SplitDeviceName("/device:GPU:3", &task, &device)); in TEST()
380 EXPECT_FALSE(DeviceNameUtils::SplitDeviceName("gpu:3", &task, &device)); in TEST()
381 EXPECT_FALSE(DeviceNameUtils::SplitDeviceName("/job:foo/task:2/replica:1", in TEST()
383 EXPECT_TRUE(DeviceNameUtils::SplitDeviceName("/device:myspecialdevice:3", in TEST()
|
| D | device_name_utils.h | 180 static bool SplitDeviceName(StringPiece name, string* task, string* device);
|
| D | device_name_utils.cc | 515 bool DeviceNameUtils::SplitDeviceName(StringPiece name, string* task, in SplitDeviceName() function in tensorflow::DeviceNameUtils
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_cache_partial.cc | 61 if (!DeviceNameUtils::SplitDeviceName(device_name, &task, &device)) { in RefreshDeviceStatus()
|
| D | device_resolver_distributed_test.cc | 103 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
|
| D | collective_param_resolver_distributed_test.cc | 98 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
|
| D | collective_rma_distributed_test.cc | 143 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
|
| D | master_session.cc | 378 CHECK(DeviceNameUtils::SplitDeviceName(node->assigned_device_name(), &task, in SplitByWorker()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer.cc | 627 DeviceNameUtils::SplitDeviceName(node->device(), &node_device, &rest); in GroupCrossDeviceControlEdges()
644 DeviceNameUtils::SplitDeviceName(input->device(), &input_device, in GroupCrossDeviceControlEdges()
691 DeviceNameUtils::SplitDeviceName(input->device(), &input_device, in GroupCrossDeviceControlEdges()
|
| D | generic_layout_optimizer.cc | 87 if (!DeviceNameUtils::SplitDeviceName(device_name, &task, &device_type) || in NumConv2DOnDeviceWithDataTypeOverThreshold()
|
| D | memory_optimizer.cc | 743 if (!DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) || in BuildSwapPair()
1266 DeviceNameUtils::SplitDeviceName(node1.device(), &task1, &device1); in CrossesTaskOrCpuGpuBoundary()
1269 DeviceNameUtils::SplitDeviceName(node2.device(), &task2, &device2); in CrossesTaskOrCpuGpuBoundary()
|
| D | arithmetic_optimizer.cc | 195 return DeviceNameUtils::SplitDeviceName(node.device(), &task, &device) && in NodeIsOnCpu()
2098 return DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) && in NodeIsOnCpuOrGpu()
|
| D | auto_mixed_precision.cc | 1096 if (DeviceNameUtils::SplitDeviceName(device_name, ¬_used, &device) && in IsOnGPU()
|
| D | layout_optimizer.cc | 591 if (DeviceNameUtils::SplitDeviceName(device_name, ¬_used, &device) && in IsOnGPU()
|
| D | generic_layout_optimizer_transposer.cc | 209 DeviceNameUtils::SplitDeviceName(device_name, &task, &device) && in ShouldProcess()
|
| D | constant_folding.cc | 3584 tensorflow::DeviceNameUtils::SplitDeviceName(input_node->device(), in MergeConcat()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 199 if (!DeviceNameUtils::SplitDeviceName(master_env_.local_devices[0]->name(), in Init()
424 if (!DeviceNameUtils::SplitDeviceName(master_env_.local_devices[0]->name(), in UpdateServerDef()
|
| D | rpc_rendezvous_mgr.cc | 221 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &call->src_worker_, in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 228 return DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) && in NodeIsOnCpu()
234 return DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) && in NodeIsOnGpu()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 716 if (!DeviceNameUtils::SplitDeviceName(d->name(), task_name, &ignored)) { in GetTaskName()
|