| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 247 .Device(edge->src()->requested_device()) in CreateEnterNodes()
268 .Device(while_op_->requested_device()) in CreateEnterNodes()
287 .Device(enter_node->requested_device()) in CreateMergeNodes()
304 cond_call_builder_.Device(while_op_->requested_device()); in CreateCondFuncCallNode()
313 .Device(while_op_->requested_device()) in CreateCondFuncCallNode()
339 .Device(merge_node->requested_device()) in CreateSwitchNodes()
356 body_call_builder_.Device(while_op_->requested_device()); in CreateBodyFuncCallNode()
373 .Device(while_op_->requested_device()) in CreateBodyFuncCallNode()
397 ->requested_device()) in CreateExitNodes()
415 .Device(while_op_->requested_device()) in CreateExitNodes()
[all …]
|
| D | lower_if_op.cc | 127 then_call_builder_.Device(if_op_->requested_device()); in CondBuilder()
132 else_call_builder_.Device(if_op_->requested_device()); in CondBuilder()
157 .Device(if_op_->requested_device()), in CreatePivotNodes()
164 .Device(if_op_->requested_device()), in CreatePivotNodes()
170 .Device(if_op_->requested_device()), in CreatePivotNodes()
199 .Device(src->requested_device()) in AddInput()
243 .Device(if_op_->requested_device()), in AddOutputs()
263 .Device(if_op_->requested_device()), in AddOutputs()
293 return builder.Device(if_op_->requested_device()) in BuildLoweredIfOutput()
|
| D | lower_case_op.cc | 112 branch_call_builders_[b].Device(case_op_->requested_device()); in CaseBuilder()
127 .Device(case_op_->requested_device()) in CreatePivotNodes()
136 .Device(case_op_->requested_device()) in CreatePivotNodes()
162 .Device(src->requested_device()) in AddInput()
212 .Device(case_op_->requested_device()) in AddOutputs()
235 .Device(case_op_->requested_device()) in AddOutputs()
263 return builder.Device(case_op_->requested_device()) in BuildLoweredCaseOutput()
|
| D | lower_while_op_test.cc | 342 ASSERT_EQ(consumer->requested_device(), gpu_0_device); in TEST()
353 ASSERT_EQ(consumer->requested_device(), gpu_0_device); in TEST()
362 ASSERT_EQ(consumer->requested_device(), gpu_0_device); in TEST()
373 ASSERT_EQ(consumer->requested_device(), gpu_0_device); in TEST()
382 ASSERT_EQ(consumer->requested_device(), gpu_0_device); in TEST()
394 ASSERT_EQ(loop_control_inputs_node->requested_device(), gpu_2_device); in TEST()
403 ASSERT_EQ(loop_executed_node->requested_device(), gpu_2_device); in TEST()
|
| D | colocation_graph.cc | 180 if (!DeviceNameUtils::ParseFullName(node.requested_device(), in SetResourceDeviceName()
183 node.requested_device(), in SetResourceDeviceName()
204 if (!DeviceNameUtils::ParseFullName(node.requested_device(), in SetRequestedDeviceName()
207 node.requested_device(), in SetRequestedDeviceName()
1211 if (DeviceNameUtils::ParseFullName(node->requested_device(), in GetDevicesForNode()
1239 " was explicitly assigned to ", node->requested_device(), in GetDevicesForNode()
1246 node->requested_device(), "' because no supported kernel for ", in GetDevicesForNode()
1255 node->requested_device(), debug_info); in GetDevicesForNode()
1264 node->requested_device(), "' because the node ", in GetDevicesForNode()
1374 ") ", node->requested_device()); in DebugInfo()
[all …]
|
| D | inline_function_utils.cc | 130 : edge->src()->requested_device(); in InputDevices()
696 << ", requested_device: " << n->requested_device() << ")"; in InlineFunctionBody()
748 << ", requested_device: " << n->requested_device() << ")"; in InlineFunctionBody()
825 .Device(caller->requested_device()) in InlineFunctionBody()
831 .Device(caller->requested_device()) in InlineFunctionBody()
|
| D | accumulate_n_optimizer.cc | 114 node_builder.Device(n->requested_device()); in RewriteIntoTempVariable()
259 .Device(n->requested_device()) in RewriteIntoAddN()
|
| D | parallel_concat_optimizer.cc | 53 node_builder.Device(n->requested_device()); in Run()
|
| D | placer.cc | 169 << node->requested_device() << "' assigned: '" in Run()
|
| D | function_utils.cc | 324 ndef->set_device(n->requested_device()); in ToGraphDef()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | collective_order.cc | 85 if (collective_nodes[i]->requested_device() != in CreateControlDependencies()
86 collective_nodes[j]->requested_device()) { in CreateControlDependencies()
93 collective_nodes[i]->requested_device()); in CreateControlDependencies()
|
| D | graph_partition_test.cc | 85 string device_name = !node->requested_device().empty() in Partition()
86 ? node->requested_device() in Partition()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | sharding_util.cc | 104 device_name = node.requested_device(); in ParseShardingFromDevice()
142 device_name = src.requested_device(); in SetShardingDeviceAssignmentFromNode()
|
| D | tf2xla_util.cc | 522 n->set_requested_device(matching_node->requested_device()); in SetNodeShardingFromNeighbors()
639 ? node->requested_device() in RewriteAssociatedFunction()
749 absl::optional<string> requested_device) { in BuildIdentityNode() argument
757 if (requested_device) { in BuildIdentityNode()
758 ndef.set_device(*requested_device); in BuildIdentityNode()
|
| D | tf2xla_util.h | 185 absl::optional<string> requested_device);
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_handle_to_key_op.cc | 36 << " on device " << ctx->op_kernel().requested_device(); in Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 49 if (DeviceNameUtils::ParseFullName(n->requested_device(), &parsed)) { in IsCpuGpuCompile()
323 VLOG(2) << "Device is " << launch->requested_device(); in BuildXlaLaunchOps()
325 def.set_device(launch->requested_device()); in BuildXlaLaunchOps()
|
| D | build_xla_ops_pass_test.cc | 67 if (n->requested_device().empty()) { in BuildXlaOps()
70 n->set_assigned_device_name(n->requested_device()); in BuildXlaOps()
|
| D | build_xla_ops_pass.cc | 152 .WithDevice(new_node->requested_device()) in MergeOutgoingDataEdges()
492 .WithDevice(n->requested_device()) in ReplaceNodeWithXlaCompileAndXlaRun()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | session_ops.cc | 52 TensorStore::TensorAndKey tk{val, id, requested_device()}; in Compute()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1083 const string& requested_device = unpadded_input->requested_device(); in SetPaddingNodesDevices() local
1085 if (!requested_device.empty() || !assigned_device.empty()) { in SetPaddingNodesDevices()
1093 out->set_requested_device(requested_device); in SetPaddingNodesDevices()
1101 out->set_requested_device(requested_device); in SetPaddingNodesDevices()
1115 return node->requested_device(); in AssignedOrRequestedDevice()
1173 node->set_requested_device(src_node->requested_device()); in PlaceOpsOnTPU()
2487 !node->requested_device().empty()) { in FindVariableInputs()
2489 TF_RET_CHECK(DeviceNameUtils::ParseFullName(node->requested_device(), in FindVariableInputs()
2493 node->set_assigned_device_name(node->requested_device()); in FindVariableInputs()
2497 node->requested_device()); in FindVariableInputs()
[all …]
|
| D | variable_merger_pass.cc | 111 node_def.set_device(handle_op->requested_device()); in MergeReadVariableOps()
|
| D | distributed_tpu_rewrite_helpers.cc | 218 string spec_string = node->requested_device(); in ForConfigurationNodeMatchingType()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 266 return node->requested_device(); in GetDeviceName()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 1039 if (node->requested_device().empty()) { in OptimizeGraph()
1047 node->set_assigned_device_name(node->requested_device()); in OptimizeGraph()
|