| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | set_device.cc | 23 Status SetDevice(const GraphDef& input_graph_def, in SetDevice() function
44 REGISTER_GRAPH_TRANSFORM("set_device", SetDevice);
|
| D | set_device_test.cc | 32 Status SetDevice(const GraphDef& input_graph_def,
90 TF_ASSERT_OK(SetDevice(graph_def, context, &result)); in TEST()
113 TF_ASSERT_OK(SetDevice(graph_def, context, &result)); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_operation.h | 62 tensorflow::Status SetDevice(const char* device);
63 void SetDevice(tensorflow::Device* device) { device_ = device; } in SetDevice() function
|
| D | eager_operation.cc | 18 tensorflow::Status EagerOperation::SetDevice(const char* device) { in SetDevice() function in tensorflow::EagerOperation
|
| D | execute.cc | 842 op->SetDevice(resource_device); in MaybeUpdateOpDevice()
886 op->SetDevice(ctx->HostCPU()); in MaybeUpdateOpDevice()
1066 op.SetDevice(device); in ExecuteSend()
1095 op.SetDevice(device); in ExecuteRecv()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session_test.cc | 364 void SetDevice(GraphDef* graph, const string& name, const string& dev) { in SetDevice() function
405 SetDevice(&def, a->name(), a_dev.name()); in TEST()
406 SetDevice(&def, b->name(), b_dev.name()); in TEST()
407 SetDevice(&def, c->name(), c_dev.name()); in TEST()
477 SetDevice(&def, fill_node->name(), cluster->devices()[0].name()); in TEST()
478 SetDevice(&def, fill_node->name(), cluster->devices()[1].name()); in TEST()
517 SetDevice(&def, a->name(), a_dev.name()); in TEST()
518 SetDevice(&def, b->name(), b_dev.name()); in TEST()
570 SetDevice(&def, a->name(), src.name()); in TEST()
571 SetDevice(&def, b->name(), dst0.name()); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | grpc_session_debug_test.cc | 215 void SetDevice(GraphDef* graph, const string& name, const string& dev) { in SetDevice() function
247 SetDevice(&def, a->name(), a_dev.name()); in TEST_F()
248 SetDevice(&def, b->name(), b_dev.name()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | get_serialized_resource_op_test.cc | 58 SetDevice(DEVICE_GPU, std::move(device)); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | cluster.h | 60 void SetDevice(const string& device) { device_ = device; } in SetDevice() function
|
| D | resolve_svdf.cc | 273 cluster->SetDevice(node.device()); in CreateCluster()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 25 void OpsTestBase::SetDevice(const DeviceType& device_type, in SetDevice() function in tensorflow::OpsTestBase
|
| D | depthwise_conv_ops_test.cc | 41 SetDevice(DEVICE_GPU, in Run()
|
| D | ops_testutil.h | 88 void SetDevice(const DeviceType& device_type, std::unique_ptr<Device> device);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 171 TF_RETURN_IF_ERROR(op->SetDevice(operation.device().c_str())); in ExecuteOp()
|
| /external/skqp/src/compute/hs/cuda/bench/ |
| D | main.c | 499 cuda(SetDevice(device)); in main()
|
| /external/skia/src/compute/hs/cuda/bench/ |
| D | main.c | 499 cuda(SetDevice(device)); in main()
|
| /external/stressapptest/src/ |
| D | worker.h | 681 virtual void SetDevice(const char *device_name);
|
| D | sat.cc | 1326 thread->SetDevice(diskfilename_[i].c_str()); in InitializeThreads()
1343 rthread->SetDevice(diskfilename_[i].c_str()); in InitializeThreads()
|
| D | worker.cc | 2605 void DiskThread::SetDevice(const char *device_name) { in SetDevice() function in DiskThread
|
| /external/toolchain-utils/android_bench_suite/ |
| D | autotest.diff | 561 +NAME = "SetDevice"
907 + set_device = ['SetDevice']
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 596 status->status = op->operation.SetDevice(device_name); in TFE_OpSetDevice()
|