| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_with_tracking_alloc_test.cc | 54 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()
179 x->set_assigned_device_name("/job:localhost/replica:0/task:0/device:SYCL:0"); in TestHWAccelerator()
184 y->set_assigned_device_name("/job:localhost/replica:0/task:0/device:GPU:0"); in TestHWAccelerator()
186 y->set_assigned_device_name("/job:localhost/replica:0/task:0/device:SYCL:0"); in TestHWAccelerator()
190 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TestHWAccelerator()
[all …]
|
| D | placer_test.cc | 737 GetNodeByName(g, "in")->set_assigned_device_name( in TEST_F()
775 GetNodeByName(g, "in")->set_assigned_device_name( in TEST_F()
919 ->set_assigned_device_name( in TEST_F()
922 ->set_assigned_device_name( in TEST_F()
1323 GetNodeByName(g, "in")->set_assigned_device_name("/job:foo"); in TEST_F()
1399 GetNodeByName(g, "in")->set_assigned_device_name("/foo:bar"); in TEST_F()
1417 GetNodeByName(g, "in")->set_assigned_device_name("/job:a"); in TEST_F()
1658 ->set_assigned_device_name("/job:a/replica:0/task:0/device:fakecpu:1"); in TEST_F()
1661 ->set_assigned_device_name("/job:a/replica:0/task:0/device:fakecpu:2"); in TEST_F()
1794 GetNodeByName(g, "id_a")->set_assigned_device_name(kFullGPU); in TEST_P()
[all …]
|
| D | direct_session_test.cc | 90 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in Initialize()
96 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
101 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in Initialize()
106 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
110 z->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
534 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TEST_F()
538 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in TEST_F()
541 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:2"); in TEST_F()
554 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in TEST_F()
660 var->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TEST()
[all …]
|
| D | graph_execution_state.cc | 242 n->set_assigned_device_name(iter->second); in RestoreStatefulNodes()
282 (*out_node)->set_assigned_device_name( in AddNode()
725 node->set_assigned_device_name(node->requested_device()); in OptimizeGraph()
|
| D | process_function_library_runtime.cc | 317 node->set_assigned_device_name(input_devices[index]); in PinArgsAndRets()
389 node->set_assigned_device_name(matching_devices[0]->name()); in PinArgsAndRets()
401 node->set_assigned_device_name(output_devices[index]); in PinArgsAndRets()
|
| D | constant_folding.cc | 556 constant_node->set_assigned_device_name(partition_device->name()); in ReplaceTensorWithConstant()
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | device_tracer_test.cc | 61 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in Initialize()
66 x->set_assigned_device_name("/job:localhost/replica:0/task:0/device:GPU:0"); in Initialize()
71 y->set_assigned_device_name("/job:localhost/replica:0/task:0/device:GPU:0"); in Initialize()
76 i->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in Initialize()
80 y_neg->set_assigned_device_name( in Initialize()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass_test.cc | 879 n->set_assigned_device_name(string(xla_gpu_device)); in TEST()
912 n->set_assigned_device_name(string(xla_gpu_device)); in TEST()
958 n->set_assigned_device_name(string(xla_gpu_dev0)); in TEST()
960 n->set_assigned_device_name(string(xla_gpu_dev1)); in TEST()
1009 n->set_assigned_device_name(string(xla_gpu_dev0)); in TEST()
1011 n->set_assigned_device_name(string(xla_gpu_dev1)); in TEST()
1040 n->set_assigned_device_name(string(xla_cpu_device)); in TEST()
1085 n->set_assigned_device_name(string(xla_cpu_device)); in TEST()
1180 n->set_assigned_device_name(xla_cpu_device); in TEST()
1207 FindNodeByName(graph.get(), "test/x")->set_assigned_device_name(kGPU0); in TEST()
[all …]
|
| D | mark_for_compilation_pass_test_helper.cc | 29 n->set_assigned_device_name(kCpuDevice); in MarkForCompilation()
|
| D | build_xla_ops_pass_test.cc | 67 n->set_assigned_device_name(kCpuDevice); in BuildXlaOps()
69 n->set_assigned_device_name(n->requested_device()); in BuildXlaOps()
|
| D | clone_constants_for_better_clustering.cc | 53 new_in->set_assigned_device_name(n->assigned_device_name()); in CloneNode()
|
| D | partially_decluster_pass_test.cc | 89 n->set_assigned_device_name(kCpuDevice); in PartiallyDecluster()
404 n->set_assigned_device_name( in TEST()
|
| D | partially_decluster_pass.cc | 131 cloned_node->set_assigned_device_name(n->assigned_device_name()); in PartiallyDeclusterNode()
|
| D | node_matchers_test.cc | 164 assigned_add.node()->set_assigned_device_name( in TEST()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session_test.cc | 611 a->set_assigned_device_name(dev_a); in TEST()
613 a_err->set_assigned_device_name(dev_a); in TEST()
615 a2->set_assigned_device_name(dev_a); in TEST()
625 b->set_assigned_device_name(dev_b); in TEST()
627 b_delay->set_assigned_device_name(dev_b); in TEST()
629 b2->set_assigned_device_name(dev_b); in TEST()
668 a->set_assigned_device_name(dev_a); in TEST()
673 a_err->set_assigned_device_name(dev_a); in TEST()
675 a2->set_assigned_device_name(dev_a); in TEST()
685 b->set_assigned_device_name(dev_b); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | subgraph.cc | 233 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
253 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
272 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
290 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
|
| D | graph_partition.cc | 350 node->set_assigned_device_name( in OptimizeControlFlowColocation()
359 node->set_assigned_device_name( in OptimizeControlFlowColocation()
376 node->set_assigned_device_name( in OptimizeControlFlowColocation()
405 res_node->set_assigned_device_name(device_name); in AddControlEnter()
418 res_node->set_assigned_device_name(device_name); in AddControlMerge()
429 res_node->set_assigned_device_name(device_name); in AddControlSwitch()
438 res_node->set_assigned_device_name(device_name); in AddControlNext()
460 res_node->set_assigned_device_name(device_name); in AddControlConst()
|
| D | graph.h | 116 void set_assigned_device_name(const string& device_name);
626 void set_assigned_device_name(Node* node, const string& device_name) { in set_assigned_device_name() function
885 inline void Node::set_assigned_device_name(const string& device_name) { in set_assigned_device_name() function
886 graph_->set_assigned_device_name(this, device_name); in set_assigned_device_name()
|
| D | mkl_tfconversion_pass.cc | 189 conversion_node->set_assigned_device_name(src->assigned_device_name()); in InsertConversionNodeOnEdge()
274 conversion_node->set_assigned_device_name(n->assigned_device_name()); in InsertInputConversionNode()
|
| D | node_builder.cc | 131 node->set_assigned_device_name(assigned_device_); in Finalize()
|
| D | mkl_layout_pass.cc | 1640 (*out)->set_assigned_device_name(orig_node->assigned_device_name()); in GetDummyMklTensorNode()
2803 new_node->set_assigned_device_name(pred->assigned_device_name()); in MergeConv2DWithBiasAdd()
2978 new_node->set_assigned_device_name(pred->assigned_device_name()); in MergePadWithConv2D()
3100 new_node->set_assigned_device_name(badd->assigned_device_name()); in MergeConv2DBackpropFilterWithBiasAddGrad()
3225 new_node->set_assigned_device_name(orig_node->assigned_device_name()); in RewriteNode()
3406 new_node->set_assigned_device_name(mklop->assigned_device_name()); in FuseTransposeMklOpTranspose()
|
| D | mkl_layout_pass_test.cc | 61 node->set_assigned_device_name(device); in InitGraph()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 237 a_node->set_assigned_device_name("foo"); in TEST()
241 a_node->set_assigned_device_name("/device:TPU_REPLICATED_CORE:2"); in TEST()
249 b_node->set_assigned_device_name("/device:TPU_REPLICATED_CORE:1"); in TEST()
|
| D | sharding_util.cc | 104 dst->set_assigned_device_name(device_name); in SetShardingDeviceAssignmentFromNode()
|
| D | tf2xla.cc | 260 node->set_assigned_device_name( in ConvertGraphToXla()
|