Home
last modified time | relevance | path

Searched refs:requested_device (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op.cc118 then_call_builder_.Device(if_op_->requested_device()); in CondBuilder()
123 else_call_builder_.Device(if_op_->requested_device()); in CondBuilder()
138 .Device(if_op_->requested_device()) in CreatePivotNodes()
144 .Device(if_op_->requested_device()) in CreatePivotNodes()
149 .Device(if_op_->requested_device()) in CreatePivotNodes()
174 .Device(src->requested_device()) in AddInput()
215 .Device(if_op_->requested_device()) in AddOutputs()
234 .Device(if_op_->requested_device()) in AddOutputs()
264 return builder.Device(if_op_->requested_device()) in BuildLoweredIfOutput()
Dlower_while_op.cc248 .Device(while_op_->requested_device()); in CreateEnterNodes()
268 .Device(while_op_->requested_device()) in CreateEnterNodes()
287 .Device(while_op_->requested_device()) in CreateMergeNodes()
303 cond_call_builder_.Device(while_op_->requested_device()); in CreateCondFuncCallNode()
312 .Device(while_op_->requested_device()) in CreateCondFuncCallNode()
341 .Device(while_op_->requested_device()) in CreateSwitchNodes()
357 body_call_builder_.Device(while_op_->requested_device()); in CreateBodyFuncCallNode()
374 .Device(while_op_->requested_device()) in CreateBodyFuncCallNode()
397 .Device(while_op_->requested_device()) in CreateExitNodes()
413 .Device(while_op_->requested_device()) in CreateExitNodes()
[all …]
Dlower_case_op.cc113 branch_call_builders_[b].Device(case_op_->requested_device()); in CaseBuilder()
128 .Device(case_op_->requested_device()) in CreatePivotNodes()
137 .Device(case_op_->requested_device()) in CreatePivotNodes()
163 .Device(src->requested_device()) in AddInput()
213 .Device(case_op_->requested_device()) in AddOutputs()
236 .Device(case_op_->requested_device()) in AddOutputs()
264 return builder.Device(case_op_->requested_device()) in BuildLoweredCaseOutput()
Dcolocation_graph.cc173 if (!DeviceNameUtils::ParseFullName(node.requested_device(), in SetResourceDeviceName()
176 node.requested_device(), in SetResourceDeviceName()
197 if (!DeviceNameUtils::ParseFullName(node.requested_device(), in SetRequestedDeviceName()
200 node.requested_device(), in SetRequestedDeviceName()
1082 if (DeviceNameUtils::ParseFullName(node->requested_device(), in GetDevicesForNode()
1110 " was explicitly assigned to ", node->requested_device(), in GetDevicesForNode()
1117 node->requested_device(), "' because no supported kernel for ", in GetDevicesForNode()
1126 node->requested_device(), debug_info); in GetDevicesForNode()
1135 node->requested_device(), "' because the node ", in GetDevicesForNode()
1245 ") ", node->requested_device()); in DebugInfo()
[all …]
Daccumulate_n_optimizer.cc114 node_builder.Device(n->requested_device()); in RewriteIntoTempVariable()
259 .Device(n->requested_device()) in RewriteIntoAddN()
Dparallel_concat_optimizer.cc57 node_builder.Device(n->requested_device()); in Run()
Dplacer.cc169 << node->requested_device() << "' assigned: '" in Run()
Dfunction.cc1517 : edge->src()->requested_device(); in InputDevices()
2060 << ", requested_device: " << n->requested_device() << ")"; in InlineFunctionBody()
2111 << ", requested_device: " << n->requested_device() << ")"; in InlineFunctionBody()
2189 .Device(caller->requested_device()) in InlineFunctionBody()
2195 .Device(caller->requested_device()) in InlineFunctionBody()
2300 ndef->set_device(n->requested_device()); in ToGraphDef()
Dgraph_execution_state.cc767 node->set_assigned_device_name(node->requested_device()); in OptimizeGraph()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util.cc92 device_name = node.requested_device(); in ParseShardingFromDevice()
102 device_name = src.requested_device(); in SetShardingDeviceAssignmentFromNode()
Dtf2xla_util.cc516 n->set_requested_device(matching_node->requested_device()); in SetNodeShardingFromNeighbors()
633 ? node->requested_device() in RewriteAssociatedFunction()
743 absl::optional<string> requested_device) { in BuildIdentityNode() argument
751 if (requested_device) { in BuildIdentityNode()
752 ndef.set_device(*requested_device); in BuildIdentityNode()
Dtf2xla_util.h185 absl::optional<string> requested_device);
/external/tensorflow/tensorflow/core/graph/
Dcollective_order.cc85 if (collective_nodes[i]->requested_device() != in CreateControlDependencies()
86 collective_nodes[j]->requested_device()) { in CreateControlDependencies()
93 collective_nodes[i]->requested_device()); in CreateControlDependencies()
Dgraph_partition_test.cc85 string device_name = !node->requested_device().empty() in Partition()
86 ? node->requested_device() in Partition()
Dgraph.h105 const string& requested_device() const;
Dgraph.cc205 const string& Node::requested_device() const { return def().device(); } in requested_device() function in tensorflow::Node
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc54 if (DeviceNameUtils::ParseFullName(n->requested_device(), &parsed)) { in IsCpuGpuCompile()
329 VLOG(2) << "Device is " << launch->requested_device(); in BuildXlaLaunchOps()
331 def.set_device(launch->requested_device()); in BuildXlaLaunchOps()
Dbuild_xla_ops_pass_test.cc67 if (n->requested_device().empty()) { in BuildXlaOps()
70 n->set_assigned_device_name(n->requested_device()); in BuildXlaOps()
Dbuild_xla_ops_pass.cc152 .WithDevice(new_node->requested_device()) in MergeOutgoingDataEdges()
488 .WithDevice(n->requested_device()) in ReplaceNodeWithXlaCompileAndXlaRun()
Dmark_for_compilation_pass.cc1004 : node->requested_device(); in BuildInitialClusterSet()
Dencapsulate_subgraphs_pass.cc459 ? node->requested_device() in MakeNodeImage()
/external/tensorflow/tensorflow/core/kernels/
Dsession_ops.cc46 TensorStore::TensorAndKey tk{val, id, requested_device()}; in Compute()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc572 } else if (!tf_node->requested_device().empty()) { in SegmentGraph()
574 tf_node->requested_device()); in SegmentGraph()
578 << tf_node->requested_device(); in SegmentGraph()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc130 if (!node->requested_device().empty()) { in GetEngineInfo()
131 device_name = node->requested_device(); in GetEngineInfo()
150 << (node->requested_device().empty() ? "assigned" : "requested") in GetEngineInfo()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.cc908 if (node->requested_device().empty()) { in OptimizeGraph()
916 node->set_assigned_device_name(node->requested_device()); in OptimizeGraph()

12