| /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/tpu/graph_rewrite/ |
| D | distributed_tpu_configuration_rewrite_pass.cc | 62 config_def.set_device(configuration_device_name); in AddConfigurationNode()
84 host_config_def.set_device(host_device_name); in AddHostConfigNode()
105 wait_def.set_device(configuration_device_name); in AddWaitNode()
132 global_tpu_array_def.set_device(host_device_name); in AddGlobalTPUArrayNode()
154 sync_def.set_device(device_name); in AddSynchronizationNode()
191 shutdown_def.set_device(shutdown_device_name); in AddShutdownNode()
218 host_disconnect_def.set_device(host_device_name); in AddHostDisconnectNode()
|
| D | incomplete_nodedef_builder.cc | 45 nodedef_.set_device(device); in Device()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 1074 fused_op.set_device(contraction.device()); in IsCpuCompatibleDataType()
1122 fused_op.set_device(contraction.device()); in IsCpuCompatibleDataType()
1175 fused_conv2d.set_device(contraction.device()); in IsCpuCompatibleDataType()
1217 fused_conv2d.set_device(contraction.device()); in IsCpuCompatibleDataType()
1260 fused_conv2d.set_device(contraction.device()); in IsCpuCompatibleDataType()
1300 contraction_node.set_device(contraction.device()); in IsCpuCompatibleDataType()
1346 fused_conv2d.set_device(contraction.device()); in IsCpuCompatibleDataType()
1398 fused_op.set_device(fused_batch_norm.device()); in IsCpuCompatibleDataType()
1423 identity_op.set_device(fused_batch_norm.device()); in IsCpuCompatibleDataType()
1467 new_shape.set_device(fused_node.device()); in IsCpuCompatibleDataType()
[all …]
|
| D | remapper_test.cc | 177 item.graph.mutable_node(i)->set_device("/device:GPU:0"); in TEST_F()
288 item.graph.mutable_node(i)->set_device("/device:GPU:0"); in TEST_F()
360 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 RunTest()
508 item.graph.mutable_node(i)->set_device("/device:GPU:0"); in TEST_F()
595 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
697 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in RunTest()
795 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
894 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
970 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
|
| D | mkl_remapper_test.cc | 123 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in FuseConv2DWithBiasAndAddNOrAdd()
266 item.graph.mutable_node(i)->set_device("/device:CPU:0"); \
380 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
493 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()
676 tmp_var->set_device(device); in SchedulingPass()
691 zeros->set_device(device); in SchedulingPass()
698 initialize->set_device(device); in SchedulingPass()
714 accumulate->set_device(device); in SchedulingPass()
786 swap_out_node->set_device(node->device()); in BuildSwapPair()
787 swap_in_node->set_device(node->device()); in BuildSwapPair()
|
| /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 | 36 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()
|
| /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 | 44 void set_device(const std::string& device) { device_ = device; } in set_device() function
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_variable_loading_test.cc | 85 saved_variable.set_device("/job:localhost/replica:0/task:0/device:CPU:1"), in TEST_P()
102 saved_variable.set_device("/job:localhost/replica:0/task:0/device:CPU:99"), in TEST_P()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | default_device.h | 33 node->set_device(device); in SetDefaultDevice()
|
| D | costmodel_test.cc | 49 device_stepstats->set_device(device_name); in GenerateStepStats()
|
| /external/tensorflow/tensorflow/core/profiler/internal/gpu/ |
| D | cupti_collector.cc | 354 sync_dev_stats->set_device( in Flush()
391 stream_dev_stats->set_device(absl::StrCat( in Flush()
397 all_streams_dev_stats->set_device( in Flush()
422 stream_dev_stats->set_device(absl::StrCat( in Flush()
429 memcpy_dev_stats->set_device( in Flush()
439 unknown_stream_dev_stats->set_device( in Flush()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 451 node.set_device("/cpu:0"); in TEST()
457 node.set_device("/gpu:0"); in TEST()
462 node.set_device(""); in TEST()
467 node.set_device("/cpu:0"); in TEST()
469 node.set_device("/gpu:0"); in TEST()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 190 operation->set_device(device); in BuildOperation()
203 input->set_device(device); in BuildOperation()
870 input.set_device(local_device_); in TEST_F()
970 input.set_device(local_device_); in TEST_F()
1018 input.set_device(local_device_); in TEST_F()
1147 remote_handle->set_device(device0); in TEST_F()
1152 lcoal_handle->set_device(device1); in TEST_F()
1158 remote_handle->set_device(device2); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_utils.cc | 322 ndef->set_device(n->assigned_device_name()); in ToGraphDef()
324 ndef->set_device(n->requested_device()); in ToGraphDef()
|
| /external/tensorflow/tensorflow/core/profiler/internal/cpu/ |
| D | host_tracer.cc | 113 dev_stats->set_device("/host:CPU"); in CollectData()
|
| /external/toolchain-utils/android_bench_suite/ |
| D | run.py | 191 def set_device(serials, remote, frequency): function
416 set_device(serials, remote, frequency)
|