| /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()
156 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TestHWAccelerator()
161 x->set_assigned_device_name("/job:localhost/replica:0/task:0/device:GPU:0"); in TestHWAccelerator()
163 x->set_assigned_device_name("/job:localhost/replica:0/task:0/device:SYCL:0"); in TestHWAccelerator()
168 y->set_assigned_device_name("/job:localhost/replica:0/task:0/device:GPU:0"); in TestHWAccelerator()
170 y->set_assigned_device_name("/job:localhost/replica:0/task:0/device:SYCL:0"); in TestHWAccelerator()
174 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TestHWAccelerator()
[all …]
|
| D | direct_session_test.cc | 65 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in Initialize()
70 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
75 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in Initialize()
80 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
237 a->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TEST_F()
241 x->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in TEST_F()
244 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:2"); in TEST_F()
257 y->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in TEST_F()
303 var->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TEST()
311 twenty_node->set_assigned_device_name( in TEST()
[all …]
|
| D | placer_test.cc | 446 GetNodeByName(g, "in")->set_assigned_device_name( in TEST_F()
486 GetNodeByName(g, "in")->set_assigned_device_name( in TEST_F()
968 GetNodeByName(g, "in")->set_assigned_device_name("/job:foo"); in TEST_F()
1044 GetNodeByName(g, "in")->set_assigned_device_name("/foo:bar"); in TEST_F()
1062 GetNodeByName(g, "in")->set_assigned_device_name("/job:a"); in TEST_F()
1271 ->set_assigned_device_name("/job:a/replica:0/task:0/device:fakecpu:1"); in TEST_F()
1274 ->set_assigned_device_name("/job:a/replica:0/task:0/device:fakecpu:2"); in TEST_F()
|
| D | graph_execution_state.cc | 229 n->set_assigned_device_name(iter->second); in RestoreStatefulNodes()
368 node->set_assigned_device_name(node->requested_device()); in OptimizeGraph()
|
| D | constant_folding.cc | 546 constant_node->set_assigned_device_name(partition_device->name()); in ReplaceTensorWithConstant()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_gateway_test.cc | 67 a->set_assigned_device_name(kDeviceName); in Initialize()
73 x->set_assigned_device_name(kDeviceName); in Initialize()
78 y->set_assigned_device_name(kDeviceName); in Initialize()
83 y_neg->set_assigned_device_name(kDeviceName); in Initialize()
525 a->set_assigned_device_name(kDeviceName); in Initialize()
528 c->set_assigned_device_name(kDeviceName); in Initialize()
535 y->set_assigned_device_name(kDeviceName); in Initialize()
628 var->set_assigned_device_name(kDeviceName); in Initialize()
639 init_val->set_assigned_device_name(kDeviceName); in Initialize()
643 init->set_assigned_device_name(kDeviceName); in Initialize()
[all …]
|
| /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/core/distributed_runtime/rpc/ |
| D | grpc_session_test.cc | 535 a->set_assigned_device_name(dev_a); in TEST()
537 a_err->set_assigned_device_name(dev_a); in TEST()
539 a2->set_assigned_device_name(dev_a); in TEST()
549 b->set_assigned_device_name(dev_b); in TEST()
551 b_delay->set_assigned_device_name(dev_b); in TEST()
553 b2->set_assigned_device_name(dev_b); in TEST()
592 a->set_assigned_device_name(dev_a); in TEST()
597 a_err->set_assigned_device_name(dev_a); in TEST()
599 a2->set_assigned_device_name(dev_a); in TEST()
609 b->set_assigned_device_name(dev_b); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 230 a_node->set_assigned_device_name("foo"); in TEST()
234 a_node->set_assigned_device_name("/device:TPU_REPLICATED_CORE:2"); in TEST()
242 b_node->set_assigned_device_name("/device:TPU_REPLICATED_CORE:1"); in TEST()
|
| D | sharding_util.cc | 105 dst->set_assigned_device_name(device_name); in SetShardingDeviceAssignmentFromNode()
|
| D | tf2xla_util.cc | 278 n->set_assigned_device_name(matching_node->assigned_device_name()); in SetNodeShardingFromNeighbors()
|
| D | tf2xla.cc | 257 node->set_assigned_device_name( in ConvertGraphToXla()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 114 void set_assigned_device_name(const string& device_name);
561 void set_assigned_device_name(Node* node, const string& device_name) { in set_assigned_device_name() function
819 inline void Node::set_assigned_device_name(const string& device_name) { in set_assigned_device_name() function
820 graph_->set_assigned_device_name(this, device_name); in set_assigned_device_name()
|
| D | graph_partition.cc | 334 node->set_assigned_device_name( in OptimizeControlFlowColocation()
343 node->set_assigned_device_name( in OptimizeControlFlowColocation()
360 node->set_assigned_device_name( in OptimizeControlFlowColocation()
389 res_node->set_assigned_device_name(device_name); in AddControlEnter()
402 res_node->set_assigned_device_name(device_name); in AddControlMerge()
413 res_node->set_assigned_device_name(device_name); in AddControlSwitch()
422 res_node->set_assigned_device_name(device_name); in AddControlNext()
444 res_node->set_assigned_device_name(device_name); in AddControlConst()
|
| D | mkl_tfconversion_pass.cc | 185 conversion_node->set_assigned_device_name(src->assigned_device_name()); in InsertConversionNodeOnEdge()
266 conversion_node->set_assigned_device_name(n->assigned_device_name()); in InsertInputConversionNode()
|
| D | subgraph.cc | 104 recv_node->set_assigned_device_name(device_info.name()); in FeedInputs()
252 send_node->set_assigned_device_name(device_info.name()); in FetchOutputs()
|
| D | mkl_layout_pass.cc | 1062 (*out)->set_assigned_device_name(orig_node->assigned_device_name()); in GetDummyMklTensorNode()
1355 (*out)->set_assigned_device_name(orig_node->assigned_device_name()); in GetDummyWorkspaceTensorNode()
1875 new_node->set_assigned_device_name(pred->assigned_device_name()); in MergeNode()
2018 new_node->set_assigned_device_name(orig_node->assigned_device_name()); in RewriteNode()
3114 (*out)->set_assigned_device_name(orig_node->assigned_device_name());
3410 (*out)->set_assigned_device_name(orig_node->assigned_device_name());
3899 new_node->set_assigned_device_name(pred->assigned_device_name());
4021 new_node->set_assigned_device_name(badd->assigned_device_name());
4132 new_node->set_assigned_device_name(orig_node->assigned_device_name());
|
| D | mkl_layout_pass_test.cc | 59 node->set_assigned_device_name(device); in InitGraph()
1920 node->set_assigned_device_name(device);
|
| D | graph.cc | 346 copy->set_assigned_device_name(node->assigned_device_name()); in CopyNode()
|
| D | graph_partition_test.cc | 87 node->set_assigned_device_name(device_name); in Partition()
|
| D | graph_constructor.cc | 553 (*node)->set_assigned_device_name(node_def.device()); in MakeNode()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_launch_ops_pass.cc | 88 launch_node->set_assigned_device_name(node->assigned_device_name()); in ReplaceNodeWithXlaLaunch()
|
| D | encapsulate_subgraphs_pass.cc | 850 sequencer_->set_assigned_device_name(device_); in MakeSequencingNode()
1030 check_op->set_assigned_device_name(device); in BuildParallelCheckOp()
1054 call_node_inputs_->set_assigned_device_name(device_); in AddFunctionCallNode()
1089 oc_subgraph->recv_at_host->set_assigned_device_name(device_); in AddRecvAtHostNode()
1132 oc_subgraph->send_from_host->set_assigned_device_name(device_); in AddSendFromHostNode()
|
| D | mark_for_compilation_pass_test.cc | 43 n->set_assigned_device_name(kCpuDevice); in MarkForCompilation()
|