Home
last modified time | relevance | path

Searched refs:set_assigned_device_name (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_with_tracking_alloc_test.cc54 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TEST()
59 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in TEST()
63 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TEST()
66 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in TEST()
172 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TestHWAccelerator()
177 x->set_assigned_device_name("/job:localhost/replica:0/task:0/device:GPU:0"); in TestHWAccelerator()
181 y->set_assigned_device_name("/job:localhost/replica:0/task:0/device:GPU:0"); in TestHWAccelerator()
184 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TestHWAccelerator()
254 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TEST()
259 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in TEST()
[all …]
Dreplicate_per_replica_nodes.cc45 replicated_node->set_assigned_device_name(device); in ReplicateNode()
143 pack_node->set_assigned_device_name(dst->assigned_device_name()); in ReplicateFromCompositeDeviceToRegularDevice()
253 n->set_assigned_device_name(allowed_devices.at(0)); in ReplicatePerReplicaNodesInFunctionGraph()
Dplacer_test.cc454 GetNodeByName(g, "n2")->set_assigned_device_name( in TEST_F()
852 GetNodeByName(g, "in")->set_assigned_device_name( in TEST_F()
907 GetNodeByName(g, "in")->set_assigned_device_name( in TEST_F()
1052 ->set_assigned_device_name( in TEST_F()
1055 ->set_assigned_device_name( in TEST_F()
1190 GetNodeByName(*g, "var")->set_assigned_device_name( in TEST_F()
1208 ->set_assigned_device_name("/job:a/replica:0/task:0/device:FakeCPU:0"); in TEST_F()
1531 GetNodeByName(g, "in")->set_assigned_device_name("/job:foo"); in TEST_F()
1657 GetNodeByName(g, "in")->set_assigned_device_name("/foo:bar"); in TEST_F()
1675 GetNodeByName(g, "in")->set_assigned_device_name("/job:a"); in TEST_F()
[all …]
Dmkl_tfconversion_pass.cc190 conversion_node->set_assigned_device_name(src->assigned_device_name()); in InsertConversionNodeOnEdge()
276 conversion_node->set_assigned_device_name(n->assigned_device_name()); in InsertInputConversionNode()
Ddirect_session_test.cc100 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in Initialize()
106 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
111 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in Initialize()
116 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
120 z->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
681 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TEST_F()
685 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in TEST_F()
688 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:2"); in TEST_F()
701 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in TEST_F()
807 var->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TEST()
[all …]
Dgraph_execution_state.cc301 n->set_assigned_device_name(iter->second); in RestoreStatefulNodes()
341 (*out_node)->set_assigned_device_name( in AddNode()
838 node->set_assigned_device_name(node->requested_device()); in OptimizeGraph()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session_test.cc646 a->set_assigned_device_name(dev_a); in TEST()
648 a_err->set_assigned_device_name(dev_a); in TEST()
650 a2->set_assigned_device_name(dev_a); in TEST()
660 b->set_assigned_device_name(dev_b); in TEST()
662 b_delay->set_assigned_device_name(dev_b); in TEST()
664 b2->set_assigned_device_name(dev_b); in TEST()
703 a->set_assigned_device_name(dev_a); in TEST()
705 a_err->set_assigned_device_name(dev_a); in TEST()
707 a2->set_assigned_device_name(dev_a); in TEST()
717 b->set_assigned_device_name(dev_b); in TEST()
[all …]
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc73 (*configuration_node)->set_assigned_device_name(configuration_device_name); in AddConfigurationNode()
94 (*host_configuration_node)->set_assigned_device_name(host_device_name); in AddHostConfigNode()
119 (*wait_node)->set_assigned_device_name(configuration_device_name); in AddWaitNode()
140 (*global_tpu_array_node)->set_assigned_device_name(host_device_name); in AddGlobalTPUArrayNode()
163 sync_node->set_assigned_device_name(device_name); in AddSynchronizationNode()
199 (*shutdown_node)->set_assigned_device_name(shutdown_device_name); in AddShutdownNode()
227 host_disconnect_node->set_assigned_device_name(host_device_name); in AddHostDisconnectNode()
Dvariable_merger_pass.cc80 node->set_assigned_device_name(device); in MergeVarHandleOps()
115 node->set_assigned_device_name(handle_op->assigned_device_name()); in MergeReadVariableOps()
Ddistributed_tpu_rewrite_pass.cc310 node->set_assigned_device_name(DeviceNameUtils::ParsedNameToString(device)); in SetNodeDeviceForTPUCommunication()
701 pad_node->set_assigned_device_name(split_node->assigned_device_name()); in CreatePadNode()
758 split_node->set_assigned_device_name(input_assigned_device); in CreateSplitNode()
1094 out->set_assigned_device_name(assigned_device); in SetPaddingNodesDevices()
1102 out->set_assigned_device_name(assigned_device); in SetPaddingNodesDevices()
1172 node->set_assigned_device_name(src_node->assigned_device_name()); in PlaceOpsOnTPU()
2082 args[i]->set_assigned_device_name(CoreDeviceLabel(*assigned_core)); in AssignArgsAndRetvalsToCores()
2183 retvals[i]->set_assigned_device_name(CoreDeviceLabel(*assigned_core)); in AssignArgsAndRetvalsToCores()
2295 shape_node->set_assigned_device_name(src->assigned_device_name()); in BuildDynamicShapeNodes()
2444 (*compile_node)->set_assigned_device_name(compile_device); in BuildCompileNode()
[all …]
Dhost_training_loop_optimization_util.cc541 reshard_op_node->set_assigned_device_name( in AddReshardOp()
595 unshard_op_node->set_assigned_device_name( in AddReshardOp()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test.cc317 ->set_assigned_device_name(xla_cpu_device); in TEST()
319 ->set_assigned_device_name(xla_cpu_device); in TEST()
321 ->set_assigned_device_name(xla_cpu_device); in TEST()
323 ->set_assigned_device_name(xla_cpu_device); in TEST()
1011 n->set_assigned_device_name(string(xla_gpu_device)); in TEST()
1044 n->set_assigned_device_name(string(xla_gpu_device)); in TEST()
1090 n->set_assigned_device_name(string(xla_gpu_dev0)); in TEST()
1092 n->set_assigned_device_name(string(xla_gpu_dev1)); in TEST()
1127 n->set_assigned_device_name(string(xla_cpu_dev0)); in TEST()
1129 n->set_assigned_device_name(string(xla_gpu_dev0)); in TEST()
[all …]
Dmark_for_compilation_pass_test_helper.cc31 n->set_assigned_device_name(kCpuDevice); in MarkForCompilation()
Dbuild_xla_ops_pass_test.cc68 n->set_assigned_device_name(kCpuDevice); in BuildXlaOps()
70 n->set_assigned_device_name(n->requested_device()); in BuildXlaOps()
Dclone_constants_for_better_clustering.cc53 new_in->set_assigned_device_name(n->assigned_device_name()); in CloneNode()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment_test.cc212 add1.node()->set_assigned_device_name("/device:CPU:0"); in TEST_F()
216 add1.node()->set_assigned_device_name(""); in TEST_F()
223 add0.node()->set_assigned_device_name(kGpu0); in TEST_F()
224 add1.node()->set_assigned_device_name(kGpu0); in TEST_F()
225 add2.node()->set_assigned_device_name(kGpu0); in TEST_F()
227 add3.node()->set_assigned_device_name(kGpu1); in TEST_F()
228 add4.node()->set_assigned_device_name(kGpu1); in TEST_F()
238 add3.node()->set_assigned_device_name(kGpuAny); in TEST_F()
239 add4.node()->set_assigned_device_name(kGpuAny); in TEST_F()
/external/tensorflow/tensorflow/core/graph/
Dsubgraph.cc232 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
252 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
271 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
289 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
Dgraph_partition.cc377 node->set_assigned_device_name( in OptimizeControlFlowColocation()
386 node->set_assigned_device_name( in OptimizeControlFlowColocation()
403 node->set_assigned_device_name( in OptimizeControlFlowColocation()
432 res_node->set_assigned_device_name(device_name); in AddControlEnter()
445 res_node->set_assigned_device_name(device_name); in AddControlMerge()
456 res_node->set_assigned_device_name(device_name); in AddControlSwitch()
465 res_node->set_assigned_device_name(device_name); in AddControlNext()
487 res_node->set_assigned_device_name(device_name); in AddControlConst()
Dgraph.h126 void set_assigned_device_name(const std::string& device_name);
677 void set_assigned_device_name(Node* node, const std::string& device_name) { in set_assigned_device_name() function
968 inline void Node::set_assigned_device_name(const std::string& device_name) { in set_assigned_device_name() function
969 graph_->set_assigned_device_name(this, device_name); in set_assigned_device_name()
Dnode_builder.cc131 node->set_assigned_device_name(assigned_device_); in Finalize()
/external/tensorflow/tensorflow/core/framework/
Drun_handler_test.cc604 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in Initialize()
610 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
615 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in Initialize()
620 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
624 z->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer_test.cc91 y->set_assigned_device_name("/device:GPU:0"); in Initialize()
96 i->set_assigned_device_name("/cpu:0"); in Initialize()
100 y_neg->set_assigned_device_name("/device:GPU:0"); in Initialize()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla.cc62 node->set_assigned_device_name( in ConvertGraphToXla()
Dtf2xla_util_test.cc239 a_node->set_assigned_device_name("foo"); in TEST()
243 a_node->set_assigned_device_name("/device:TPU_REPLICATED_CORE:2"); in TEST()
252 b_node->set_assigned_device_name("/device:TPU_REPLICATED_CORE:1"); in TEST()
Dsharding_util.cc144 dst->set_assigned_device_name(device_name); in SetShardingDeviceAssignmentFromNode()

12