| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_placer_test.cc | 45 node.set_device("CPU"); in TEST()
50 node.set_device("GPU:0"); in TEST()
74 node.set_device("CPU"); in TEST()
78 node.set_device("GPU:0"); in TEST()
105 node.set_device("/device:TPU:0"); in TEST()
130 node.set_device("/device:CPU:0"); in TEST()
133 node.set_device("/device:GPU:0"); in TEST()
152 node.set_device("/device:CPU:0"); in TEST()
212 node.set_device(device); in TEST()
304 node.set_device("/job:my_job/replica:0/task:0/cpu:0"); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 919 fused_op.set_device(contraction.device()); in AddFusedContractionNode()
963 fused_op.set_device(contraction.device()); in AddFusedContractionNode()
1012 fused_conv2d.set_device(contraction.device()); in AddFusedConv2DNode()
1054 fused_conv2d.set_device(contraction.device()); in AddFusedConv2DNode()
1097 fused_conv2d.set_device(contraction.device()); in AddFusedConv2DNode()
1138 fused_conv2d.set_device(contraction.device()); in AddFusedContractionNode()
1175 fused_conv2d.set_device(contraction.device()); in AddFusedContractionNode()
1227 fused_op.set_device(fused_batch_norm.device()); in AddFusedBatchNormExNode()
1252 identity_op.set_device(fused_batch_norm.device()); in AddFusedBatchNormExNode()
1295 new_shape.set_device(fused_node.device()); in AddBatchNormNodes()
[all …]
|
| D | remapper_test.cc | 175 item.graph.mutable_node(i)->set_device("/device:GPU:0"); in TEST_F()
286 item.graph.mutable_node(i)->set_device("/device:GPU:0"); in TEST_F()
358 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
419 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
497 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
575 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
652 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
745 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
816 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
|
| D | mkl_remapper_test.cc | 68 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
141 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
|
| D | memory_optimizer.cc | 359 new_trigger_node->set_device(original_recomputed_node->device()); in AddRecomputeControlDependencyNodes()
421 copied_node->set_device(original_node->device()); in RecomputeSubgraph()
671 tmp_var->set_device(device); in SchedulingPass()
686 zeros->set_device(device); in SchedulingPass()
693 initialize->set_device(device); in SchedulingPass()
709 accumulate->set_device(device); in SchedulingPass()
781 swap_out_node->set_device(node->device()); in BuildSwapPair()
782 swap_in_node->set_device(node->device()); in BuildSwapPair()
|
| D | constant_folding.cc | 236 added_node->set_device(node.device()); in AddControlDependency()
507 added_node->set_device(shape_n_node->device()); in MaterializeShapes()
676 out[j]->set_device(node.device()); in MaterializeBroadcastGradientArgs()
790 reduction_indices->set_device(node->device()); in MaterializeReductionIndices()
1379 const_out->set_device(node->device()); in FoldMergeNode()
1392 const_index->set_device(node->device()); in FoldMergeNode()
1494 added_node->set_device(node->device()); in FoldNode()
1833 const_node->set_device(node->device()); in ReplaceBinaryOperationWithBroadcastTo()
2327 axis_node->set_device(node->device()); in SimplifyPack()
2378 new_node->set_device(node->device()); in MoveConstantsPastEnter()
[all …]
|
| D | shape_optimizer.cc | 126 size_node.set_device(node.device()); in Optimize()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | remove_device_test.cc | 44 mul_node1->set_device("//cpu:0"); in TestRemoveDevice()
53 add_node2->set_device("//device:GPU:1"); in TestRemoveDevice()
|
| D | set_device_test.cc | 43 mul_node1->set_device("/device:CPU:0"); in CreateDeviceGraph()
52 add_node2->set_device("/device:GPU:1"); in CreateDeviceGraph()
|
| D | remove_device.cc | 37 new_node->set_device(""); in RemoveDevice()
|
| D | set_device.cc | 37 new_node->set_device(new_device); in SetDevice()
|
| D | flatten_atrous.cc | 110 flattened_conv_node.set_device(conv_node.device()); in FlattenAtrousConv()
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_svdf_test.cc | 85 node->set_device(""); in AddNewNode()
98 node->set_device(""); in AddNewNode()
121 node->set_device(""); in AddShapeNode()
|
| D | resolve_svdf.cc | 106 svdf_node->set_device(device_); in CreateNodes()
170 merged_node->set_device(const_node_parts[0]->device()); in MaybeMergeConstNodes()
|
| D | resolve_cluster.cc | 40 new_node->set_device(node.device()); in AddNodeToGraph()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_handle.cc | 31 proto->set_device(device()); in AsProto()
44 set_device(proto.device()); in FromProto()
|
| D | resource_handle.h | 42 void set_device(const string& device) { device_ = device; } in set_device() function
|
| D | function_testlib.cc | 55 n.set_device(device); in NDef()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 421 node.set_device("/cpu:0"); in TEST()
427 node.set_device("/gpu:0"); in TEST()
432 node.set_device(""); in TEST()
437 node.set_device("/cpu:0"); in TEST()
439 node.set_device("/gpu:0"); in TEST()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | default_device.h | 33 node->set_device(device); in SetDefaultDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 180 operation->set_device(device); in AddOperationToEnqueueRequest()
187 input->set_device(device); in AddOperationToEnqueueRequest()
578 input.set_device(local_device_); in TEST_F()
626 input.set_device(local_device_); in TEST_F()
|
| D | cluster_function_library_runtime.cc | 159 remote_op->set_device(function_data->target); in Run()
|
| /external/toolchain-utils/android_bench_suite/ |
| D | run.py | 191 def set_device(serials, remote, frequency): function
416 set_device(serials, remote, frequency)
|
| D | autotest.diff | 803 diff --git a/site_utils/set_device.py b/site_utils/set_device.py
807 +++ b/site_utils/set_device.py
878 + Entry point for set_device script.
907 + set_device = ['SetDevice']
909 + autotest_path, argv, set_device, arguments.remote,
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 45 def.set_device(DEVICE_CPU); in TestBitcastOp()
|