| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 203 .Device(while_op_->requested_device()) in CreateEnterNodes()
220 .Device(while_op_->requested_device()) in CreateEnterNodes()
236 .Device(while_op_->requested_device()) in CreateMergeNodes()
247 cond_call_builder_.Device(while_op_->requested_device()); in CreateCondFuncCallNode()
256 .Device(while_op_->requested_device()) in CreateCondFuncCallNode()
278 .Device(while_op_->requested_device()) in CreateSwitchNodes()
289 body_call_builder_.Device(while_op_->requested_device()); in CreateBodyFuncCallNode()
306 .Device(while_op_->requested_device()) in CreateBodyFuncCallNode()
320 .Device(while_op_->requested_device()) in CreateExitNodes()
332 ib.Device(while_op_->requested_device()); in CreateExitNodes()
[all …]
|
| D | lower_if_op.cc | 115 then_call_builder_.Device(if_op_->requested_device()); in CondBuilder()
116 else_call_builder_.Device(if_op_->requested_device()); in CondBuilder()
127 .Device(if_op_->requested_device()) in CreatePivotNodes()
133 .Device(if_op_->requested_device()) in CreatePivotNodes()
138 .Device(if_op_->requested_device()) in CreatePivotNodes()
163 .Device(src->requested_device()) in AddInput()
204 .Device(if_op_->requested_device()) in AddOutputs()
220 .Device(if_op_->requested_device()) in AddOutputs()
276 ib.Device(if_op_->requested_device()); in BuildLoweredIfOutput()
|
| D | colocation_graph.cc | 191 if (!DeviceNameUtils::ParseFullName(node.requested_device(), in SetRequestedDeviceName()
194 node.requested_device(), in SetRequestedDeviceName()
748 if (DeviceNameUtils::ParseFullName(node->requested_device(), in GetDevicesForNode()
777 "was explicitly assigned to ", node->requested_device(), in GetDevicesForNode()
784 node->requested_device(), "' because no supported kernel for ", in GetDevicesForNode()
791 node->requested_device(), debug_info); in GetDevicesForNode()
798 node->requested_device(), "' because the node ", in GetDevicesForNode()
898 ") ", node->requested_device()); in DebugInfo()
977 if (!node.requested_device().empty()) { in InitializeMember()
982 node.requested_device(), node.type_string(), false, member)); in InitializeMember()
|
| D | parallel_concat_optimizer.cc | 57 node_builder.Device(n->requested_device()); in Run()
|
| D | placer.cc | 108 << node->requested_device() << "' assigned: '" in Run()
|
| D | accumulate_n_optimizer.cc | 82 node_builder.Device(n->requested_device()); in rewriteNode()
|
| D | graph_execution_state.cc | 725 node->set_assigned_device_name(node->requested_device()); in OptimizeGraph()
|
| D | process_function_library_runtime.cc | 298 return &node.requested_device(); in AssignedOrRequestedDeviceName()
|
| /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()
|
| D | graph.h | 103 const string& requested_device() const;
|
| D | graph.cc | 189 const string& Node::requested_device() const { return def().device(); } in requested_device() function in tensorflow::Node
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | sharding_util.cc | 92 device_name = node.requested_device(); in ParseShardingFromDevice()
102 device_name = src.requested_device(); in SetShardingDeviceAssignmentFromNode()
|
| D | tf2xla_util.cc | 523 n->set_requested_device(matching_node->requested_device()); in SetNodeShardingFromNeighbors()
640 ? node->requested_device() in RewriteAssociatedFunction()
750 absl::optional<string> requested_device) { in BuildIdentityNode() argument
758 if (requested_device) { in BuildIdentityNode()
759 ndef.set_device(*requested_device); in BuildIdentityNode()
|
| D | tf2xla_util.h | 184 absl::optional<string> requested_device);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 53 if (DeviceNameUtils::ParseFullName(n->requested_device(), &parsed)) { in IsCpuGpuCompile()
328 VLOG(2) << "Device is " << launch->requested_device(); in BuildXlaLaunchOps()
330 def.set_device(launch->requested_device()); in BuildXlaLaunchOps()
|
| D | build_xla_ops_pass_test.cc | 66 if (n->requested_device().empty()) { in BuildXlaOps()
69 n->set_assigned_device_name(n->requested_device()); in BuildXlaOps()
|
| D | build_xla_ops_pass.cc | 329 .WithDevice(n->requested_device()) in ReplaceNodeWithXlaCompileAndXlaRun()
|
| D | mark_for_compilation_pass.cc | 1118 : node->requested_device(); in RunImpl()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 568 } else if (!tf_node->requested_device().empty()) { in SegmentGraph()
570 tf_node->requested_device()); in SegmentGraph()
574 << tf_node->requested_device(); in SegmentGraph()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | session_ops.cc | 46 TensorStore::TensorAndKey tk{val, id, requested_device()}; in Compute()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 781 if (node->requested_device().empty()) { in OptimizeGraph()
788 node->set_assigned_device_name(node->requested_device()); in OptimizeGraph()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 216 auto node_device = node->requested_device(); in GetEngineInfo()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 154 const string& requested_device() const; // Same as def().device()
|
| D | op_kernel.cc | 130 const string& OpKernel::requested_device() const { return def_->device(); } in requested_device() function in tensorflow::OpKernel
|