| /external/tensorflow/tensorflow/core/util/ |
| D | device_name_utils_test.cc | 369 TEST(DeviceNameUtilsTest, SplitDeviceName) { in TEST() argument
372 EXPECT_TRUE(DeviceNameUtils::SplitDeviceName( in TEST()
376 EXPECT_TRUE(DeviceNameUtils::SplitDeviceName( in TEST()
381 DeviceNameUtils::SplitDeviceName("/device:GPU:3", &task, &device)); in TEST()
384 EXPECT_FALSE(DeviceNameUtils::SplitDeviceName("gpu:3", &task, &device)); in TEST()
385 EXPECT_FALSE(DeviceNameUtils::SplitDeviceName("/job:foo/task:2/replica:1", in TEST()
387 EXPECT_TRUE(DeviceNameUtils::SplitDeviceName("/device:myspecialdevice:3", in TEST()
|
| D | device_name_utils.h | 189 static bool SplitDeviceName(StringPiece name, std::string* task,
|
| D | device_name_utils.cc | 520 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 | collective_param_resolver_distributed_test.cc | 101 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
|
| D | collective_rma_distributed_test.cc | 154 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
|
| D | master_session.cc | 380 CHECK(DeviceNameUtils::SplitDeviceName(node->assigned_device_name(), &task, in SplitByWorker()
1320 DeviceNameUtils::SplitDeviceName(devices_->client_device()->name(), in CreateWorkerSessions()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer.cc | 649 DeviceNameUtils::SplitDeviceName(node->device(), &node_device, &rest); in GroupCrossDeviceControlEdges()
666 DeviceNameUtils::SplitDeviceName(input->device(), &input_device, in GroupCrossDeviceControlEdges()
713 DeviceNameUtils::SplitDeviceName(input->device(), &input_device, in GroupCrossDeviceControlEdges()
|
| D | memory_optimizer.cc | 748 if (!DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) || in BuildSwapPair()
1276 DeviceNameUtils::SplitDeviceName(node1.device(), &task1, &device1); in CrossesTaskOrCpuGpuBoundary()
1279 DeviceNameUtils::SplitDeviceName(node2.device(), &task2, &device2); in CrossesTaskOrCpuGpuBoundary()
|
| D | generic_layout_optimizer.cc | 87 if (!DeviceNameUtils::SplitDeviceName(device_name, &task, &device_type) || in NumConvOnDeviceWithDataTypeOverThreshold()
|
| D | arithmetic_optimizer.cc | 197 return DeviceNameUtils::SplitDeviceName(node.device(), &task, &device) && in NodeIsOnCpu()
2134 return DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) && in NodeIsOnCpuOrGpu()
|
| D | auto_mixed_precision.cc | 1128 if (DeviceNameUtils::SplitDeviceName(device_name, ¬_used, &device) && in IsOnDevice()
|
| D | layout_optimizer.cc | 592 if (DeviceNameUtils::SplitDeviceName(device_name, ¬_used, &device) && in IsOnGPU()
|
| D | generic_layout_optimizer_transposer.cc | 272 DeviceNameUtils::SplitDeviceName(device_name, &task, &device) && in ShouldProcess()
|
| D | constant_folding.cc | 3825 tensorflow::DeviceNameUtils::SplitDeviceName(input_node->device(), in MergeConcat()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 156 return DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) && in NodeIsOnCpu()
162 return DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) && in NodeIsOnGpu()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 206 if (!DeviceNameUtils::SplitDeviceName(master_env_.local_devices[0]->name(), in Init()
447 if (!DeviceNameUtils::SplitDeviceName(master_env_.local_devices[0]->name(), in UpdateServerDef()
|
| D | rpc_rendezvous_mgr.cc | 243 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &call->src_worker_, in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 983 if (!DeviceNameUtils::SplitDeviceName(d->name(), task_name, &ignored)) { in GetTaskName()
|