Home
last modified time | relevance | path

Searched refs:set_device (Results 1 – 25 of 75) sorted by relevance

123

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_placer_test.cc45 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 …]
Danalytical_cost_estimator.cc60 node->set_device(op_context.device_name); in AddCostNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc501 fused_conv2d->set_device(matched.conv2d->device()); in AddFusedConv2DNode()
528 fused_conv2d->set_device(matched.conv2d->device()); in AddFusedConv2DNode()
558 fused_conv2d->set_device(matched.conv2d->device()); in AddFusedConv2DNode()
587 fused_conv2d->set_device(matched.conv2d->device()); in AddFusedConv2DNode()
614 fused_conv2d->set_device(matched.conv2d->device()); in AddFusedConv2DNode()
648 new_shape->set_device(fused_node.device()); in AddBatchNormNodes()
663 reshaped_scale->set_device(fused_node.device()); in AddBatchNormNodes()
674 reshaped_offset->set_device(fused_node.device()); in AddBatchNormNodes()
685 reshaped_mean->set_device(fused_node.device()); in AddBatchNormNodes()
696 reshaped_variance->set_device(fused_node.device()); in AddBatchNormNodes()
[all …]
Dremapper_test.cc135 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
199 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
274 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
352 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
424 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
Dmemory_optimizer.cc339 new_trigger_node->set_device(original_recomputed_node->device()); in AddRecomputeControlDependencyNodes()
401 copied_node->set_device(original_node->device()); in RecomputeSubgraph()
645 tmp_var->set_device(device); in SchedulingPass()
660 zeros->set_device(device); in SchedulingPass()
667 initialize->set_device(device); in SchedulingPass()
683 accumulate->set_device(device); in SchedulingPass()
755 swap_out_node->set_device(node->device()); in BuildSwapPair()
756 swap_in_node->set_device(node->device()); in BuildSwapPair()
Dconstant_folding.cc274 added_node->set_device(node->device()); in AddControlDependency()
477 added_node->set_device(shape_n_node->device()); in MaterializeShapes()
646 out[j]->set_device(node.device()); in MaterializeBroadcastGradientArgs()
760 reduction_indices->set_device(node->device()); in MaterializeReductionIndices()
1275 const_out->set_device(node->device()); in FoldMergeNode()
1288 const_index->set_device(node->device()); in FoldMergeNode()
1382 added_node->set_device(node->device()); in FoldNode()
2311 axis_node->set_device(node->device()); in SimplifyPack()
2357 new_node->set_device(node->device()); in MoveConstantsPastEnter()
2417 false_node->set_device(node->device()); in SimplifySwitch()
[all …]
Dlayout_optimizer.cc430 node->set_device(device_name); in AddNodePermConst()
454 node->set_device(device_name); in AddNodeConstScalar()
854 node->set_device(node_->device()); in AddNodeTranspose()
969 added_node->set_device(node_->device()); in AddNodeDataFormatOp()
1345 node->set_device(node_->device()); in AddNodeShapeConst()
1375 node->set_device(node_->device()); in AddNodeReshape()
/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_svdf_test.cc85 node->set_device(""); in AddNewNode()
98 node->set_device(""); in AddNewNode()
121 node->set_device(""); in AddShapeNode()
Dresolve_svdf.cc106 svdf_node->set_device(device_); in CreateNodes()
170 merged_node->set_device(const_node_parts[0]->device()); in MaybeMergeConstNodes()
Dresolve_cluster.cc40 new_node->set_device(node.device()); in AddNodeToGraph()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dremove_device_test.cc44 mul_node1->set_device("//cpu:0"); in TestRemoveDevice()
53 add_node2->set_device("//device:GPU:1"); in TestRemoveDevice()
Dset_device_test.cc43 mul_node1->set_device("/device:CPU:0"); in CreateDeviceGraph()
52 add_node2->set_device("/device:GPU:1"); in CreateDeviceGraph()
Dremove_device.cc37 new_node->set_device(""); in RemoveDevice()
Dset_device.cc37 new_node->set_device(new_device); in SetDevice()
Dflatten_atrous.cc110 flattened_conv_node.set_device(conv_node.device()); in FlattenAtrousConv()
/external/tensorflow/tensorflow/core/framework/
Dresource_handle.cc31 proto->set_device(device()); in AsProto()
39 set_device(proto.device()); in FromProto()
Dresource_handle.h40 void set_device(const string& device) { device_ = device; } in set_device() function
Dfunction_testlib.cc55 n.set_device(device); in NDef()
/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc397 node.set_device("/cpu:0"); in TEST()
399 node.set_device("/gpu:0"); in TEST()
403 node.set_device(""); in TEST()
408 node.set_device("/cpu:0"); in TEST()
410 node.set_device("/gpu:0"); in TEST()
/external/tensorflow/tensorflow/core/graph/
Ddefault_device.h33 node->set_device(device); in SetDefaultDevice()
/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op_test.cc42 def.set_device(DEVICE_CPU); in TestBitcastOp()
/external/toolchain-utils/android_bench_suite/
Drun.py191 def set_device(serials, remote, frequency): function
416 set_device(serials, remote, frequency)
Dautotest.diff803 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/
Dkernels_test.cc74 def.set_device(device_name); in GetFakeKernel()
/external/e2fsprogs/ext2ed/
Dmain.c120 set_device(buf); in main()

123